Machine Learning Software Engineer, Research
Core
Building AI-driven simulation software for engineering and manufacturing, enabling high-fidelity multi-physics simulation to optimize design, manufacturing, and operations for advanced industries.
Role type
Senior IC machine learning software engineer (research & simulation)
Builds
AI-driven simulation software stack for engineering and manufacturing
Domain
Deep-tech, numerical physics, aerospace & defense, materials, energy, semiconductors, automotive
Deliverable
production ML models
Required skills
deep learning, probabilistic methods, scientific computing, high-performance computing (CPU/GPU clusters), parallelised/distributed training, cloud computing (AWS/Azure/GCP), Python (PyTorch/JAX), C/C++, software engineering best practices, containerization/orchestration (Docker/Kubernetes/Slurm)
Preferred skills
federated learning, distributed computing frameworks (Spark/Dask), high-performance computing frameworks (MPI/OpenMP/CUDA/Triton), MLOps
Technologies
PyTorch, JAX, NumPy, SciPy, Pandas, AWS, Azure, GCP, Docker, Kubernetes, Slurm, Spark, Dask, MPI, OpenMP, CUDA, Triton
Responsibilities
Design and optimize scalable machine learning models for physics and engineering problems; implement distributed training architectures for multi-node/multi-GPU training; transform prototype models into robust implementations; work with research scientists to scale foundation models; identify and integrate best libraries and tools; own research work-streams; translate research results into reusable libraries and products; mentor less experienced colleagues.
Seniority
Senior, hands-on IC with mentorship responsibilities