AI-Driven Simulation Platform Engineer

GREEN14; Stockholms kommun; On-site; SEK 900,000 - 1,300,000; Full time

Job summary

GREEN14 in Stockholm develops simulation technology by combining multiphysics models, AI and data to automate industrial processes. You will turn research code into reliable software and own the technical foundations of the simulation platform, collaborating with scientists and engineers.

You will build robust backend systems, data pipelines and ML tooling, ensuring testable, scalable software that can be deployed across projects and industries.

Qualifications

• You have experience taking technically complex software from prototype to production.
• You might come from software engineering, ML engineering, scientific computing or data engineering.
• Experience with Python and modern software engineering.
• Experience with JAX or other numerical and ML frameworks.
• Experience with backend architectures and APIs.
• Experience with data pipelines and databases.
• Experience with ML infrastructure and MLOps.
• Experience with testing, CI/CD and reproducible workflows.
• Experience with computationally intensive workloads.

Responsibilities

• 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 our products mature.

Skills

Python

JAX

Backend architectures

APIs

Data pipelines

ML infrastructure

CI/CD

Docker

Kubernetes

GPU computing

FastAPI

ML tooling

Tools

Docker

Kubernetes

FastAPI

JAX

Job description

GREEN14 in Stockholm develops simulation technology by combining multiphysics models, AI and data to automate industrial processes. You will turn research code into reliable software and own the technical foundations of the simulation platform, collaborating with scientists and engineers.

You will build robust backend systems, data pipelines and ML tooling, ensuring testable, scalable software that can be deployed across projects and industries.