AI Researcher
BeyondMathGenerative Physics company
London, United KingdomSenior
Insight Partners
Acrew Capital
Cambridge Innovation Capital
UP.Partners
InMotion Ventures
Builders VC
Data & AI
About the role
TL;DR
Lead the research and development of novel ML architectures for solving complex PDEs.
- •As a Senior AI Researcher, you will lead the research and development of novel ML architectures designed to solve complex PDEs including aerodynamic simulations.
- •Key Responsibilities Architect Physics-AI Foundations: Lead the research and development of novel ML architectures designed to solve complex PDEs including aerodynamic simulations.
- •Bridge Research & Production: Translate high-level mathematical concepts into clean, high-performance code.
- •Advance Geometry Representation: Pioneer new ways to represent complex geometric design variations for efficient use in deep learning models.
- •Strategic Leadership: Mentor junior researchers and engineers.
- •Help define our internal research standards, reproducibility pipelines, and HPC infrastructure requirements.
- •External Impact: Represent BeyondMath in the global AI community.
- •Publish influential research at top-tier conferences.
- •Cross-Functional Collaboration: Partner with CFD specialists and software engineers to ensure our models respect physical constraints while maintaining the speed advantages of neural networks.
- •Requirements PhD or MSc in Computer Science, Physics, Mathematics, or a related quantitative field. 5+ years of post-grad experience in AI/ML research, with a demonstrable track record of models made it from the lab into production environments.
- •Deep Technical Mastery: Expert-level proficiency in PyTorch, JAX, or TensorFlow, with a focus on building custom layers, loss functions, and optimization loops.
- •Published Excellence: A strong record of high-quality publications in top-tier venues.
- •Systems Thinking: Experience with scalable training infrastructure, including distributed training across GPU clusters and data pipeline automation.
Required skills
PyTorchJAXTensorFlowC++
Domain expertise
aerospaceautomotivecleantech
Tech stack
PyTorchJAXTensorFlowC++