Machine Learning Engineer (Physics-based)

Root Access; New York (NY); On-site; USD 130,000 - 190,000; Full time

Job summary

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.

Qualifications

• Experience with Python for scientific/engineering ML applications.
• Ability to work with ML models in physics-informed contexts.
• Experience in building and evaluating ML models for engineering problems.
• Experience automating large-scale simulations and data pipelines.

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.
• Own reproducible pipelines for data generation, model training, evaluation, and deployment.

Skills

Python

ML frameworks

Education

Physics/Math/CS/EE degree

Tools

PyTorch

JAX

Job description

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.