About the job

PhysicsX is seeking a Machine Learning Software Engineer (Research) to join their hybrid team in London. The company, rooted in numerical physics and Formula One, develops AI-driven simulation software for engineering and manufacturing across industries like aerospace, energy, semiconductors, and automotive. This role involves working closely with research scientists and simulation engineers to design, build, and optimize scalable machine learning models for real-world physics and engineering problems. Responsibilities include developing distributed training architectures for multi-node/multi-GPU systems, implementing and transforming prototype models into production-ready software, scaling foundation models using cloud or on-premise services, and contributing to reusable ML libraries and tools. Candidates should have an MSc or PhD in computer science, machine learning, or related fields, with strong experience in high-performance and distributed computing (e.g., MPI, CUDA, Spark, Dask), model scaling, Python libraries (PyTorch, JAX), and software best practices like version control, CI/CD, and MLOps. PhysicsX offers competitive compensation, equity, flexible hybrid working, personal development opportunities, generous leave, office perks, and supports diversity through initiatives like sponsoring women in STEM. The company is a sponsor of UK Skilled Worker visas.