Senior Machine Learning Engineer (Large Systems)
$260,400–$352,200 year
On-siteGdańsk, Pomerania, Poland
Gdańsk, Pomerania, PolandOn-sitePart Time$260,400–$352,200 yearSenior LevelAI HardwareMedium
Part TimeSenior LevelMediumAI Hardware
Job Summary
Implement latest machine learning models and optimise them for performance and accuracy, scaling to thousands of accelerators. Benchmark models and key ML techniques to identify performance bottlenecks, then design and conduct experiments on novel AI methods to evaluate results. Test and evaluate new internal software releases, provide feedback to engineering teams, make necessary code fixes, and conduct code reviews. Collaborate with Research, Software, and Product teams to define, build, and test next-generation AI hardware while engaging with the AI community to track latest developments.
Required Qualifications
- Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field
- Proficiency in deep learning frameworks like PyTorch/JAX
- Strong Python or C++ software development skills
- Expertise in deep learning from model training to optimisation and evaluation
- Experience in distributed training or inference of ML models across 64+ accelerators
- Capable of designing, executing and reporting from ML experiments
- Developed deep understanding of performance bottlenecks and how to overcome them
- Ability to move quickly in a dynamic environment
- Enjoy cross-functional work collaborating with other teams
- Strong communicator - able to explain complex technical concepts to different audiences
Desired Qualifications
- Experience in one or more of: MLOps for Kubernetes-based clusters
- Building production systems with large language models
- Efficient computing based on low-precision arithmetic
- Experience writing C++/Triton/CUDA kernels for performance optimisation of ML models
- Familiarity with HPC systems and networking including Infiniband, NVLink, RoCE technologies
- Have contributed to open-source projects or published research papers in relevant fields
- Knowledge of cloud computing platforms
- Keen to present, publish and deliver talks in the AI community
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.