Machine Learning Software Engineer, Research

PhysicsX

London, United Kingdom
Equity options available; 10% employer pension con...
On-site
Msc or phd in computer science or physics
Distributed training for foundation models
Python with pytorch or jax frameworks
PhysicsX is building an AI-driven simulation software stack to accelerate hardware innovation across aerospace, energy, and automotive industries

Job Summary

  • PhysicsX is building an AI-driven simulation software stack to accelerate hardware innovation across aerospace, energy, and automotive industries.
  • The role involves designing scalable machine learning models and implementing distributed training architectures for multi-node and multi-GPU environments.
  • Employees enjoy a flat hierarchy, equity options, enhanced parental leave, and a hybrid work model blending office time in Shoreditch with remote days.

Matching Summary

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

Salary

Equity options available; 10% employer pension contribution; Free office lunches

Skills & Requirements

Must-have

  • MSc or PhD in Computer Science or Physics
  • Distributed training for foundation models
  • Python with PyTorch or JAX frameworks
  • High-performance computing on GPU clusters
  • Scaling ML models for engineering domains

Nice-to-have

  • Experience with federated learning
  • C/C++ for computer vision or geometry
  • Strong mentorship and collaboration skills
  • Cloud platform experience (AWS, Azure, GCP)
  • Background in numerical physics or F1

Key Requirements

  • MSc or PhD in CS, ML, Physics, or related field
  • >2 years professional data-driven role experience
  • Experience with MPI, OpenMP, CUDA, or Triton
  • Knowledge of Docker, Kubernetes, and Slurm

Work Rights

Not specified

Tailored Resume

Cover Letter