Staff Machine Learning Software Engineer, Research

PhysicsX

London, United Kingdom
Equity options available; 10% employer pension con...
On-site
Distributed training architectures multi-gpu
Python pytorch jax deep learning frameworks
High-performance computing cpu gpu clusters
PhysicsX is building an AI-driven simulation software stack to accelerate hardware innovation across advanced industries like aerospace and automotive

Job Summary

  • PhysicsX is building an AI-driven simulation software stack to accelerate hardware innovation across advanced industries like aerospace and automotive.
  • The role involves shaping research strategy, defining technical direction, and delivering scalable machine learning models that address real-world physics problems.
  • Employees enjoy a flat hierarchy, equity options, enhanced parental leave, and a hybrid work model blending office collaboration with remote flexibility.

Matching Summary

PhysicsX is building an AI-driven simulation software stack to accelerate hardware innovation across advanced industries like aerospace and automotive.

Salary

Equity options available; 10% employer pension contribution; Free office lunches; Enhanced parental leave; Private medical insurance; Wellhub subscription; Eye tests; Personal development support

Skills & Requirements

Must-have

  • Distributed training architectures multi-GPU
  • Python PyTorch JAX deep learning frameworks
  • High-performance computing CPU GPU clusters
  • Cloud platforms AWS Azure GCP
  • Scientific computing and numerical physics
  • MLOps CI/CD containerization Docker Kubernetes

Nice-to-have

  • Federated learning capacity experience
  • C/C++ for computer vision geometry
  • Mentoring junior colleagues professional growth
  • Strategic influence on research culture
  • Strong communication with customers stakeholders

Key Requirements

  • MSc or PhD in CS ML Physics Engineering
  • 4 years professional industry data-driven experience
  • Experience scaling foundation models at large scale

Work Rights

Not specified

Tailored Resume

Cover Letter