AI-enhanced for better readability
Senior Machine Learning Engineer – LLM & Reinforcement Learning (m/f/d)
Source: hackernews
Compensation: €160.000+
Type: Full-Time
Location: ONSITE
About the Role
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