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

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 https://warmwind.com/careers/senior-machine-learning-enginee...