Machine Learning Engineer

Molex; Fremont (CA); On-site; USD 170,000 - 250,000; Full time

Benefits offered by this job

medical

dental

vision

flexible spending and health savings

life insurance

ADD

disability

retirement

paid vacation/time off

educational assistance

infertility assistance

paid parental leave

adoption assistance

Job summary

Molex in the United States seeks an ML Engineer to build physics-informed surrogate models on Azure ML that predict engineering outcomes directly from design parameters. They will pre-screen candidate designs in milliseconds so only the most promising ones require full high-fidelity simulation, accelerating the design-optimization cycle.

You will join the platform team behind our Azure AI/ML engineering tools, collaborating with data scientists, LLM engineers, and MLOps to keep GPU-heavy

Qualifications

• Hands-on experience building, training, and deploying ML models in production.
• Experience with physics-informed ML or surrogate modeling for engineering systems.
• Strong Python programming with PyTorch or TensorFlow.
• Familiar with cloud ML platforms and model deployment concepts.

Responsibilities

• Design and train surrogate models on Azure GPU compute.
• Incorporate physics-informed constraints to ensure valid predictions.
• Build model-uncertainty and confidence scoring to guide full-simulation validation.
• Deploy and version models via Azure ML endpoints and model registry; monitor drift.
• Benchmark surrogate performance against full simulations to guide tuning.

Skills

Python

PyTorch

TensorFlow

Azure ML

uncertainty quantification

_graph neural networks

Tools

Azure Machine Learning

ND/NC series GPUs

model registry

Azure ML endpoints

Job description

Your Job

The ML Engineer will build physics-informed surrogate models on Azure Machine Learning that predict engineering simulation outcomes directly from design parameters. They will be pre-screening candidate designs in milliseconds so only the most promising ones require full high-fidelity simulation, accelerating the design-optimization cycle.

Our Team

Established in 1938, Molex delivers comprehensive electronic solutions for various markets, including data communications, telecommunications, consumer electronics, industrial, automotive, commercial vehicle, aerospace and defense, medical, and lighting. You'll join the platform team behind our Azure AI/ML engineering tools, partnering closely with data scientists, LLM engineers, and MLOps teams to keep GPU-heavy training and simulation workloads reliable and fast.

What You Will Do

• Design and train surrogate models (neural networks, Gaussian processes, gradient-boosted trees, GNNs/PINNs) on Azure GPU compute (ND/NC series).
• Incorporate physics-informed constraints so predictions stay physically valid, not just statistically fit.
• Build model-uncertainty and confidence scoring to decide which designs need full simulation validation, then retrain as new results arrive.
• Deploy and version models via Azure ML endpoints and model registry; monitor for drift on a rolling basis.
• Benchmark surrogate vs. full-simulation speedup to guide platform-level performance tuning.

Who You Are (Basic Qualifications)

• Extensive hands-on experience building, training, and deploying ML models in production - not just using pretrained APIs.
• 10+ years building ML for physical/engineering systems (surrogate modeling, physics-informed ML, or scientific ML).
• Strong Python with PyTorch or TensorFlow.
• Understanding of relevant engineering/physics fundamentals and simulation data formats for your domain.
• Experience with Azure Machine Learning or a similar cloud ML platform.
• Familiarity with uncertainty quantification (Bayesian approaches, ensembling).

What Will Put You Ahead

• Direct experience with industry-standard EM or physics simulation tools.
• Geometric deep learning (graph neural networks, mesh-based models) for CAD data.
• Background in RF/high-speed electronics or interconnect design.

For this role, we anticipate paying $170,000 - $250,000 per year. This role is eligible for variable pay, issued as a monetary bonus or in another form.

At Koch companies, we are entrepreneurs. This means we openly challenge the status quo, find new ways to create value and get rewarded for our individual contributions. Any compensation range provided for a role is an estimate determined by available market data. The actual amount may be higher or lower than the range provided considering each candidate's knowledge, skills, abilities, and geographic location. If you have questions, please speak to your recruiter about the flexibility and detail of our compensation philosophy.

Hiring Philosophy

All Koch companies value diversity of thought, perspectives, aptitudes, experiences, and backgrounds. We are Military Ready and Second Chance employers. Learn more about our hiring philosophy here.

Who We Are

As a Koch company, Molex is a leading supplier of connectors and interconnect components, driving innovation in electronics and supporting industries from automotive to health care and consumer to data communications. The thousands of innovators who work for Molex have made us a global electronics leader. Our experienced people, groundbreaking products and leading-edge technologies help us deliver a wider array of solutions to more markets than ever before.

At Koch, employees are empowered to do what they do best to make life better. Learn how our business philosophy helps employees unleash their potential while creating value for themselves and the company.

Our Benefits

Our goal is for each employee, and their families, to live fulfilling and healthy lives. We provide essential resources and support to build and maintain physical, financial, and emotional strength - focusing on overall wellbeing so you can focus on what matters most. Our benefits plan includes - medical, dental, vision, flexible spending and health savings accounts, life insurance, ADD, disability, retirement, paid vacation/time off, educational assistance, and may also include infertility assistance, paid parental leave and adoption assistance. Specific eligibility criteria is set by the applicable Summary Plan Description, policy or guideline and benefits may vary by geographic region. If you have questions on what benefits apply to you, please speak to your recruiter.

• medical
• dental
• vision
• flexible spending and health savings accounts
• life insurance
• ADD
• disability
• retirement
• paid vacation/time off
• educational assistance
• infertility assistance
• paid parental leave
• adoption assistance

Equal Opportunities

Equal Opportunity Employer, including disability and protected veteran status. Except where prohibited by state law, some offers of employment are conditioned upon successfully passing a drug test. This employer uses E-Verify. Please click here for additional information. (For Illinois E-Verify information click here, aqui, or tu).