Machine Learning Engineer

Root, Inc.
New York, United States
26 days ago
Apply on startup.jobs
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Job source

Tech stack

Python (Programming Language) Machine Learning Tensorflow Surrogate Models Information Technology Engineering Base Data Generation

Job description

  • 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.

Requirements

  • 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.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on startup.jobs
Prepare application

Good distractions

Loading talks and stories from around this role…