7 months ago
London, United KingdomSenior
Responsibilities
- Collaborate with simulation engineers and research scientists to understand and solve physics and engineering challenges.
- Transform prototype implementations into robust production-grade ML models, libraries, and tooling.
- Build and operate multi-GPU and multi-node distributed systems for model training and inference at petabyte and hundreds-of-GPU scale.
- Optimize models and pipelines from job scheduling through GPU kernels.
- Design and build scalable end-to-end machine learning infrastructure on CoreWeave and AWS.
- Apply engineering best practices and coach colleagues in adopting them.
Requirements
- Demonstrated software engineering practices including versioning, testing, API design, MLOps, and CI/CD.
- Experience building ML models and pipelines in Python using PyTorch, TensorFlow, or JAX.
- Experience with GPU programming and optimizing models across the stack, including CUDA, Triton, or Warp kernels.
- Experience training and serving machine learning models at scale.
- Experience with large-scale data processing.
- Ability to work autonomously, adapt quickly, and scope, plan, and deliver effectively.
- Strong collaboration and communication skills, particularly with research scientists.
Benefits
- Hybrid work model combining Shoreditch office time with work-from-home days.
- Equity options.
- 10% employer pension contribution.
- Free office lunches.
- Enhanced parental leave, including three months full-pay paternity leave and six months full-pay maternity leave.
- YellowNest nursery scheme.
- 25 days of annual leave plus public holidays.
- Private medical insurance with 100% employee cover.
- Wellhub subscription, eye tests, Employee Assistance Programme, personal development support, Bike2Work scheme, season ticket loan, and Octopus EV salary sacrifice.
Categories
About PhysicsX
PhysicsX builds an AI-driven simulation software stack that enables high-fidelity, multi-physics modeling and optimization for engineering and manufacturing teams. The company sells software and delivery services to enterprises in aerospace and defense, automotive, semiconductors, materials, and energy. It is privately held and headquartered in London, with offices in London and New York.