English-speaking JAX jobs in Germany

LLM Application Engineer

Germany, Germany, Germany, Germany

Remote Full-time

Strong software engineering fundamentals, experience building AI-powered applications, hands-on work with LLMs, generative AI or agent systems, prompt and workflow design, clean production-quality code, and a bias toward shipping …

LLM, Python, OpenAI API, PyTorch, JAX

Senior Deep Learning Compiler Engineer - PyTorch

Berlin

Full-time Senior

8+ YOE

PyTorch, CUDA, Python, JAX

AI Software Development Engineer - Neuromorphic Computing (m/f/d) (Fixed-Term)

München, Bayern

Hybrid Full-timeFixed-term Senior

5+ YOE Bachelor's degree with 5+ years of experience, Master's degree with 3+ years of experience, or PhD in: Computer ScienceComputer EngineeringElectrical EngineeringApplied MathematicsPhysicsOr a related discipline Strong …

C++, Python, Git, PyTorch, TensorFlow, JAX

Member of Technical Staff, Machine Learning

Germany, Germany, Germany, Germany

Remote Full-time Lead

Python, PyTorch, JAX

Platform Engineer

Berlin

Hybrid Full-time

Python, Kubernetes, MLOps, CI/CD, AWS, Google Cloud

Research Software Engineer - Scientific Computing & GPU Infrastructure

München, Bayern

Full-time Relocation support

A degree in Computer Science, Physics, Applied Mathematics, Engineering or a related field, or equivalent practical experience. Strong Python software engineering skills and a track record of independently delivering maintainable …

Python, PyTorch, JAX, CUDA, Linux, AWS

Machine Learning Engineer (m/f/d) VLA Training & Evaluation

München, Bayern

Full-time

Agile, PyTorch, JAX

Member of Technical Staff (m/f/d)

Stuttgart, Baden-Württemberg

On-site Full-time Lead Relocation support

Academic Excellence: PhD or Master’s degree in Computer Science, Machine Learning, Robotics, or a related field, with a focus on Deep Learning and Computer Vision. Proven Research Track Record: Significant publications and …

Computer Vision, PyTorch, JAX, LLM

Software Engineer - Simulation Platform

München, Bayern

On-site Full-time

We are looking for a software practitioner with a user-first mindset, interested in tackling a diverse mix of challenges: from the optimization of scientific computing workflows to building cloud- and web-based systems for data …

Python, React, TypeScript, FastAPI, JAX

AIML Engineer, AI for Science

Heidelberg, Baden-Württemberg

Hybrid Full-time

If you have the following characteristics, it would be a plus: Experience in design, development and deployment of commercial AI/ML software. Experience with Large Language Models (LLMs) and Agentic AI (e.g., tool use, …

LLM, Agile, PyTorch, JAX, TensorFlow, Python

AIML Software Engineer, AI for Science

Heidelberg, Baden-Württemberg

Hybrid Full-time

CI/CD, Agile, Python, FastAPI, Google Cloud, Azure

Senior Scientific Software Engineer — Simulation and Machine Learning

Germany

Hybrid Full-time Senior

Experience PhD or equivalent in physics, applied mathematics or computational science. Strong scientific computing in Julia, Python, JAX or comparable. Real numerical depth: ODE solvers, optimisation, stochastic processes, …

Julia, Python, JAX

PhD Position: World Models for Software Architecture Synthesis (Constructor Fabric)

Bremen

Thesis / research Senior

3+ YOE Python (professional level). Type hints, packaging, `pytest`, profiling, `asyncio`, memory-aware data structures. You write unbuffered, logged, restartable long-running jobs by default. Training loops written from scratch, custom …

Python, pytest, MLflow, PyTorch, Ray, Dask

AI Scientist - Physics Models

München, Bayern

Full-time Relocation support

PhD or Master's in CS/AI or an engineering science: Mechanical Engineering, Electrical Engineering, Computational Fluid Dynamics, Structural Mechanics, EDA, Semiconductor Engineering, or a related field Strong, hands-on machine …

Python, Linux, JAX

Applied AI/ML Engineer (Agents)

Berlin

Hybrid Full-time Mid

4+ YOE

PyTorch, JAX, CI/CD, LLM

Senior/Principal Engineer – Power Electronics (AI-Driven Designs for Future AI Data Centers)

Nürnberg, Bayern

Full-time Lead

R, PyTorch, TensorFlow, JAX

Robotics Software Engineer - Grasping

München, Bayern

Full-time Senior

5+ YOE

C++, Python, JAX, TensorFlow, PyTorch

Group Lead AI Engineering (human)

Metzingen / Riederich, Riederich, Baden-Wurttemberg, Germany, Baden-Württemberg

On-site Full-time Lead

7+ YOE

MLOps, Python, C++, PyTorch, JAX, AWS

Senior ML Research Scientist

Berlin

Full-time Senior

5+ YOE

PyTorch, TensorFlow, JAX, Python, NLP

Senior ML Research Developer

Berlin

Full-time Senior

5+ YOE

PyTorch, TensorFlow, JAX, AWS, Google Cloud, Azure

Senior Tensor AI Scientist

Germany - Hybrid

Hybrid Full-time Senior

5+ YOE Required Must have more than 5 years of relevant post-PhD experience (or equivalent experience after an MSc) with significant relevant experience in machine learning, applied mathematics, computer science, or a closely related …

PyTorch, JAX, Python, C++