Machine Learning Engineer

• Build backend architectures connecting sensor data, physics models, ML models, and simulation workflows
• Turn JAX-based research code into maintainable, production-ready software
• Build pipelines connecting experimental data, simulation, surrogates, and optimization
• Develop infrastructure for model training, evaluation, and reproducibility
• Build APIs, packages, and repositories for long-term production use
• Establish testing, CI/CD, versioning, and monitoring for scientific software
• Identify computational bottlenecks and improve performance
• Contribute to deployment and cloud infrastructure as products mature
• Work directly with scientists and engineers to turn research code into reliable software
• Own substantial technical foundations of the simulation platform

Machine Learning Engineer jobs in Sweden

Simulation Platform – ML Engineer