Machine Learning Engineer

5V Tech; New York (NY); On-site; USD 140,000 - 210,000; Full time

Job summary

5V Tech is seeking a highly skilled Scientific Machine Learning Engineer to design, develop and train advanced ML models for physics and engineering problems. You will build scalable data pipelines and collaborate across software, ML and hardware teams to deliver production-ready AI solutions.

Ideal candidates hold a Master's or PhD in a quantitative field, with 4+ years' experience in PyTorch or JAX, and strong Python and numerical computing skills using NumPy/SciPy, Open3D and related

Qualifications

• Master's or PhD in a highly quantitative field (CS, math, EE, physics).
• 4+ years of commercial experience with PyTorch or JAX.
• Experience with Scientific Machine Learning models and PDE-based learning.
• Strong Python skills with NumPy/SciPy; experience with geometric/spatial datasets (Open3D, Shapely).

Responsibilities

• Design, develop and train state-of-the-art Scientific Machine Learning models.
• Build high-performance data pipelines for complex engineering datasets.
• Work with PCB/CAD data formats.
• Integrate graph-based and transformer-based AI with physics-aware learning.
• Optimize large-scale training and inference for GPUs.
• Collaborate with software, ML and hardware engineers to deploy production-ready AI solutions.

Skills

Python programming

PyTorch

JAX

Numerical optimization

Education

Master's or PhD in a quantitative field

Tools

PyTorch

JAX

Open3D

Shapely

Job description

Help build the next generation of AI models that understand real-world physics.

I'm partnering with a well-funded, venture-backed technology startup developing advanced AI-powered hardware solutions. They're looking for a Scientific Machine Learning Engineer to help build cutting-edge machine learning models capable of solving complex physics and engineering problems.

This is a rare opportunity to join a highly technical team working at the intersection of Scientific Machine Learning, deep learning, computational physics and electronic design automation (EDA).

You'll play a key role in developing advanced AI models that learn physical behaviour, creating scalable data pipelines and collaborating with multidisciplinary engineers to deliver next-generation engineering intelligence.

What You'll Be Doing

• Design, develop and train state-of-the-art Scientific Machine Learning models.
• Build high-performance data pipelines capable of processing complex engineering and geometric datasets.
• Work with industry-standard electronic design file formats, including PCB and CAD data.
• Collaborate on integrating graph-based and transformer-based AI models with physics-aware learning systems.
• Optimise large-scale training and inference workloads for GPU environments.
• Work closely with software, machine learning and hardware engineers to develop production-ready AI solutions.

What We're Looking For

• Master\'s or PhD in Computer Science, Mathematics, Electrical Engineering, Physics or another highly quantitative discipline.
• 4+ years of commercial experience with PyTorch or JAX.
• Hands-on experience developing Scientific Machine Learning models such as:
• Other neural operators or PDE-based learning approaches
• Strong mathematical background including:
• Partial Differential Equations (PDEs)
• Automatic Differentiation
• Numerical Optimisation (Adam, L-BFGS or similar)
• Strong Python skills with experience working on geometric or spatial datasets using libraries such as NumPy, SciPy, Shapely, Open3D or similar.

Interested?

If you're passionate about Scientific Machine Learning, computational physics or next-generation AI systems, I'd love to speak with you. Apply directly or get in touch for a confidential discussion. Company details will be shared with shortlisted candidates.

Take the next step in your career and apply today!

5V Tech are acting as an Employment Agency for the purposes of this job vacancy. We offer a reward scheme if you can recommend someone for this position, up to $250 for you and an additional $250 to a charity of your choice, 5V Tech are recognised talent solutions experts within IoT and Deep Tech working across Europe, the UK, and North America.