Machine Learning Software Engineer, Research at PhysicsX | England | Rezi

Machine Learning Software Engineer, Research at PhysicsX

Machine Learning Software Engineer, Research

PhysicsX · England

1 months ago

Machine Learning Software Engineer, Research

PhysicsX · England

2 months ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

PhysicsX is an AI-driven simulation software company focused on accelerating hardware innovation. We are building a simulation software stack for engineering and manufacturing across advanced industries, enabling high-fidelity, multi-physics simulation through AI inference to unlock new levels of optimization and automation.

Responsibilities

  • Work closely with research scientists and simulation engineers to build and deliver models addressing real-world physics and engineering problems.
  • Design, build, and optimize machine learning models with a focus on scalability and efficiency.
  • Transform prototype model implementations into robust and optimized versions.
  • Implement distributed training architectures for multi-node/multi-GPU training and explore federated learning capacity using cloud and on-premise services.
  • Design, build, and scale foundation models for science and engineering, optimizing model training for large data and multi-GPU cloud compute.
  • Identify the best libraries, frameworks, and tools for modeling efforts.
  • Own Research work-streams at different levels, depending on seniority.
  • Discuss work results and implications with colleagues and customers, focusing on addressing real-world problems.
  • Translate research results into reusable libraries, tooling, and products at the intersection of data science and software engineering.
  • Foster a nurturing environment and mentor colleagues with less experience in ML/Engineering.

Requirements

  • Enthusiasm for developing machine learning solutions, especially deep learning and/or probabilistic methods, and associated supporting software solutions for science and engineering.
  • Ability to work autonomously, scope, and effectively deliver projects across various domains.
  • Strong problem-solving skills with the ability to analyze issues, identify causes, and recommend solutions quickly.
  • Excellent collaboration and communication skills with teams and customers.
  • MSc or PhD in computer science, machine learning, applied statistics, mathematics, physics, engineering, software engineering, or a related field.
  • Record of experience in scientific computing.
  • Record of experience in high-performance computing (CPU / GPU clusters).
  • Record of experience in parallelized / distributed training for large / foundation models.
  • Ideally >2 years of experience in a data-driven role in a professional setting.
  • Exposure to scaling and optimizing ML models, training and serving foundation models at scale (federated learning a bonus).
  • Exposure to distributed computing frameworks (e.g., Spark, Dask) and high-performance computing frameworks (MPI, OpenMP, CUDA, Triton).
  • Exposure to cloud computing (on hyper-scaler platforms, e.g., AWS, Azure, GCP).
  • Experience building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications.
  • Experience with C/C++ for computer vision, geometry processing, or scientific computing.
  • Experience with software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps).
  • Experience with containerization and orchestration (Docker, Kubernetes, Slurm).
  • Experience writing pipelines and experiment environments, including running experiments in pipelines systematically.

Skills

  • Deep learning
  • Probabilistic methods
  • Machine learning
  • Distributed training
  • Federated learning
  • Scientific computing
  • High-performance computing
  • Distributed computing frameworks
  • Cloud computing
  • Python
  • PyTorch
  • JAX
  • C/C++
  • Software engineering
  • Containerization
  • Orchestration

Location

  • Shoreditch office

Work Type

  • Hybrid

Experience Level

  • Multiple levels
  • 2 years

Education Level

  • MSc
  • PhD

Benefits

  • Equity options
  • 10% employer pension contribution
  • Free office lunches
  • Enhanced parental leave (3 months full pay paternity, 6 months full pay maternity)
  • YellowNest nursery scheme
  • 25 days of Annual Leave (+ Public Holidays)
  • Private medical insurance (100% employee cover)
  • Wellhub Subscription
  • Eye tests
  • Personal development support
  • Employee Assistance Programme (EAP)
  • Bike2Work scheme
  • Season ticket loan
  • Octopus EV salary sacrifice

About the Company

  • PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software.
  • We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries.
  • Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive.

Equal Opportunity

  • We value diversity and are committed to equal employment opportunity regardless of sex, race, religion, ethnicity, nationality, disability, age, sexual orientation or gender identity.
  • We strongly encourage individuals from groups traditionally underrepresented in tech to apply.
  • To help make a change, we sponsor bright women from disadvantaged backgrounds through their university degrees in science and mathematics.
  • We collect diversity and inclusion data solely for the purpose of monitoring the effectiveness of our equal opportunities policies and ensuring compliance with UK employment and equality legislation.
  • This information is confidential, used only in aggregate form, and will not influence the outcome of your application.