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

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

Salary: €160.000+

Employment Type: Full-Time

Location: Onsite in Jena, Germany

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

Language requirements:

• German or English