Senior Machine

Senior Machine Learning Engineer – LLM & Reinforcement Learning (m/f/d)

€160.000+ | Full-Time | ONSITE

As a Senior Machine Learning Engineer at warmwind your work will drive our next-gen AI models, shaping the future of machine intelligence beyond traditional paradigms. You'll work with massive compute clusters (500+ H100 GPUs) and cutting-edge reinforcement learning techniques to create highly efficient, scalable, and groundbreaking AI systems.

Responsibilities:

Design, train, and optimize Large Language Models (LLMs) from scratch

Scale distributed training on massive GPU clusters (500+ H100 GPUs)

Implement advanced reinforcement learning techniques (RLHF, adversarial self-play, real-time control)

Develop high-performance architectures for multi-modal AI systems

Build simulation environments for RL-based AI agents

Optimize inference speed and efficiency for real-world deployment

Your Profile:

Deep expertise in LLMs – you’ve built and trained large-scale models yourself

Experience with large-scale distributed training on 500+ GPU superclusters

Deep understanding of reinforcement learning, neural network optimization, and self-play methods

Expert in PyTorch, TensorFlow, JAX & low-level optimization techniques (CUDA, Triton, DeepSpeed, etc.)

Familiarity with high-performance computing (HPC, NVLink, InfiniBand, parallel computing)

Strong publication track record in AI/ML research is a plus

Relocation to Jena, Germany after initial onboarding

Language requirements:

German or English

warmwind.com/careers/senior-machine-learning-enginee...