
Warmwind is hiring a
Senior Machine Learning Engineer
About Warmwind
Warmwind builds next-generation AI models, including large language models and reinforcement learning-powered systems, focusing on scalable, efficient AI deployments.
Job Description
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 agents. Optimize inference speed and deployment efficiency. 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. Expertise in PyTorch, TensorFlow, JAX and 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 onboarding. Language requirements: German or English.Location
Jena, Germany
Salary
160K
Benefits
Not Specified
Tech Tags
CUDADeepSpeedDistributed ComputingHpcInfinibandJaxNvLinkPyTorchReinforcement LearningRlhfTensorFlowTritonmultimodalself-play
Senior Role
Date Listed
03 March, 2026 (15 days ago)
Loading...








