Role overview
- Developing custom neural network architectures using PyTorch
- Building and curating high-quality training datasets
- Performing data augmentation, feature extraction and labelling
- Running structured training, validation and optimisation cycles
- Converting models for efficient inference using tools such as ONNX or similar
- Working with quantised models for performance-constrained environments
- Collaborating closely with software engineers on integration
- Communicating technical work clearly to both technical and non-technical stakeholders
- You may also contribute to broader Python-based software development activities.
Basic qualifications
- Degree in Computer Science, Engineering, Mathematics or related discipline
- Strong Python development capability
- Strong C or C++ skills
- Deep understanding of neural network architectures such as RNNs and Transformers
- Experience building and training models using pure PyTorch
- Experience designing and managing training pipelines
- Experience with model optimisation and conversion for deployment
- Strong understanding of dataset preparation and feature engineering
- Excellent communication and technical writing ability
Benefits
- Company pension
- On-site parking
Tags & focus areas
Used for matching and alerts on DevFound Fulltime Ai Machine Learning Deep Learning Data Science Pytorch