Machine Learning Performance Engineer

Job description

Keysight is at the forefront of technology innovation, delivering breakthroughs and trusted insights in electronic design, simulation, prototyping, test, manufacturing, and optimization. Our ~15,000 employees create world-class solutions in communications, 5G, automotive, energy, quantum, aerospace, defense, and semiconductor markets for customers in over 100 countries. Learn more about what we do.

Our award-winning culture embraces a bold vision of where technology can take us and a passion for tackling challenging problems with industry-first solutions. We believe that when people feel a sense of belonging, they can be more creative, innovative, and thrive at all points in their careers.

The AI Models and Data Science team at Keysight AI Labs is hiring a ML Performance Engineer to make our training and inference stacks as fast as the math allows. You'll own end-to-end performance: profiling training workloads on multi-GPU clusters, writing custom CUDA kernels and LibTorch C++ extensions for hot paths, and optimizing inference for embedding in production software where every millisecond matters.

This role sits at the intersection of ML, systems engineering, and HPC. You'll work directly with MLEs and data scientists driving the modeling work, and with the engineering teams shipping these models into Keysight products.

• Profile and optimize training workloads — multi-GPU scaling efficiency, throughput, memory footprint, mixed precision, gradient checkpointing tradeoffs
This is an excerpt. Read the full job description on Keysight Technologies careers →