Machine Learning Engineer (Physics-based)

Role Description

Root Access is hiring a Machine Learning Engineer to develop models that predict and optimize the behavior of electronic systems. You will build automated simulation pipelines, generate training datasets, and develop models grounded in electrical, thermal, and electromagnetic physics. Working across our hardware and machine learning teams, you will turn engineering simulations and physical measurements into tools that accelerate electronics development.

Responsibilities

  • Build and scale automated workflows for electrical, thermal, and electromagnetic simulations.

  • Develop surrogate models that approximate physics simulations with faster inference.

  • Create training datasets from simulations and hardware measurements.

  • Incorporate physical constraints and engineering knowledge into model architectures, training, and evaluation.

  • Validate model predictions against established solvers and experimental results.

  • Develop active learning and optimization workflows to identify promising designs and prioritize simulations.

  • Own reproducible pipelines for data generation, model training, evaluation, and deployment.

Qualifications

  • Strong Python skills and experience with PyTorch, JAX, or similar ML frameworks.

  • Experience developing and evaluating machine learning models for scientific or engineering applications.

  • Working knowledge of numerical methods and physics-based modeling.

  • Experience automating simulation tools or computational solvers.

  • Ability to build reliable software pipelines and work with large engineering datasets.

  • Degree in physics, applied mathematics, computer science, electrical engineering, or a related field, or equivalent experience.

  • Preferred: PhD with experience in simulations, surrogate models, neural operators, or physics-informed machine learning.

Summary

Build automated simulation workflows and surrogate models that approximate electrical, thermal and electromagnetic physics with faster inference. Create training datasets from simulations and hardware measurements and validate predictions against solvers. Requires strong Python and experience with PyTorch, JAX or similar ML frameworks, plus numerical methods and physics-based modeling.

Responsibilities

Build automated simulation workflows; develop surrogate models; create training datasets from simulations and measurements; validate predictions against solvers and experimental results; develop active learning and optimization workflows; own reproducible pipelines for data generation, training, evaluation and deployment

Qualifications

Strong Python skills; experience with PyTorch, JAX or similar ML frameworks; experience developing and evaluating ML models for scientific or engineering applications; working knowledge of numerical methods and physics-based modeling; experience automating simulation tools or computational solvers

Education requirements

Degree in physics, applied mathematics, computer science, electrical engineering, or a related field, or equivalent experience