Principal Machine Learning Infrastructure Engineer

Physicsx · London

In short

▸Ingeniero principal de infraestructura de ML que construye y opera pipelines de entrenamiento y servicio a escala para modelos físicos de IA.
▸Trabajas diariamente con científicos de investigación y ML engineers en un entorno de simulación de alta fidelidad con datos de mallas grandes.
▸Destacado: Trabajas con hardware NVIDIA DGX B200 y modelos como Transolver y Point Cloud Transformer.

N/A

In ~1 minute you get: who interviews you, the likely questions answered from your CV, and your CV tailored to this job. Free, no card.

What they ask for

✓Experiencia comprobada en infraestructura de entrenamiento distribuido de modelos de IA.
✓Conocimiento profundo de pipelines de entrenamiento con optimización de throughput y tolerancia a fallos.
✓Capacidad para resolver cuellos de botella en I/O de datos con conjuntos de mallas grandes.
✓Experiencia en implementación de sistemas de observabilidad y seguimiento de experimentos.
✓Habilidades sólidas en colaboración con científicos de investigación y equipos de ML.
✓Conocimiento en modelado y despliegue de modelos con reproducibilidad y capacidad de fine-tuning.

Don't tick every box? That's normal — your free dossier shows your gaps and how to cover them in the interview.

NVIDIA DGX B200Neural Operator ArchitecturesTransolverPoint Cloud TransformerDistributed TrainingCheckpointingGradient AccumulationMulti-node SynchronizationExperiment TrackingObservability Systems

Who should you write to at Physicsx?

Your free dossier identifies the people who'd interview you — their background, what they value, and how to reach out so you stand out before applying.

About us PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across the entire engineering lifecycle, PhysicsX unlocks new levels of optimization and automation in design, manufacturing, and operations — empowering engineers to push the boundaries of possibility. Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive. Note: We are currently recruiting for multiple positions, however please only apply for the role that best aligns with your skillset and career goals. The Role The Principal ML Infrastructure Engineer will extend and operate the infrastructure that powers our research model training, fine-tuning, and serving pipelines. You will be embedded within our Research function, partnering directly with ML engineers and research scientists to ensure they can train Large Physics Models efficiently and reliably at scale. Team Context In this role, you will be vertically embedded in Research, working daily with: • Research Scientists who determine the model architectures and methods • ML Engineers who implement and develop the models • Simulation Data Engineers who are accountable for upstream data pipelines You will have end-to-end responsibilities over the research infrastructure, with the autonomy to make architectural decisions and the responsibility to keep data flowing reliably. Horizontally, you will be part of an infrastructure engineering group responsible for infrastructure across the company. What you will do Training Infrastructure • Design and operate distributed training infrastructure for neural operator architectures (Transolver, Point Cloud Transformer, etc.) on our large NVIDIA DGX B200 platform. • Optimize training pipelines for throughput, fault tolerance, and cost efficiency, including checkpointing strategies, gradient accumulation, and multi-node synchronization. • Build and maintain experiment tracking and observability systems that give researchers clear visibility into training runs, hyperparameter sweeps, and model performance. Data I/O and Performance • Solve data loading bottlenecks for large-scale mesh datasets. • Optimize data pipelines for efficient I/O from cloud storage, including prefetching, caching, and format optimization. • Work with heterogeneous data sources of varying formats and resolutions. Model Serving and Deployment • Build serving infrastructure for pre-trained LPMs, supporting both zero-shot inference and uncertainty quantification (Monte Carlo Dropout). • Design and implement model packaging pipelines for customer deployment. Models must run reliably in customer environments with fine-tuning capabilities. • Ensure reproducibility: any model checkpoint should be deployable with consistent behaviour. Platform and Tooling • Improve developer experience for the Research team with fast iteration cycles, reliable CI/CD, clear debugging tools. • Collaborate with the broader Infrastructure team on shared patterns and standards. What you bring to the table • Ability to scope and effectively deliver projects, prioritising activity as needed. • Problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly. • Excellent collaboration and communication skills, especially in a res

Don't apply unprepared

We research who's interviewing you, tailor your CV and rehearse you live — first one free.