Role overview
You will design, train and optimise high-performance machine learning models and integrate them into production environments.
What you'll work on
- 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.
What we're looking for
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
Tags & focus areas
Used for matching and alerts on DevFound Fulltime Ai Machine Learning Deep Learning Data Science Pytorch