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...
ATS Match is available
1) Upload your resume. 2) Open any job and click Check ATS Match to see your fit score.
Sign in to check your resume match