Computational Scientist / Scientific Machine Learning

Fusionality is building the measurement, simulation, and control stack for the fusion energy reactors of the future. Spun out of EPFL’s Swiss Plasma Center and backed by top investors, our technology enables reliable operation, real-time monitoring, and analysis of high-temperature plasmas in fusion devices. We are founded by world-class experts at the intersection of fusion physics, advanced control, and AI (ex-Google DeepMind, EPFL).

We are assembling a founding engineering team in Lausanne to build our core technology platform. We are looking for a Computational Scientist specialized in Scientific Machine Learning (SciML) to join our team. This job is for you if you know how to write numerical solvers that are fast, correct and differentiable end to end, and want to see them run against real plasma data and inside real control systems.

We are looking for a PhD-level or senior computational researcher for this role

Responsibilities

• Design, implement and validate differentiable solvers in JAX for plasma physics problems such as free-boundary equilibrium, transport and state estimation.
• Make them robust and fast enough for production use via advanced numerical techniques for differential equation time stepping, as well as optimization of GPU/TPU performance.
• Using automatic differentiation for model calibration against experimental data, parameter estimation and controller design and optimization, working with our data and control engineers.
• Code ownership: quality, testing, verification against reference solutions, packaging and documentation