17 days ago
Remote, Germany +10 moreStaff+
Responsibilities
- Convert stakeholder and business requirements into technical specifications.
- Train LLMs from scratch on a large GPU cluster.
- Collect and process pre-training and fine-tuning datasets.
- Design, deploy, support, and improve production ML systems and existing subsystems.
- Take ownership of an entire subsystem and make independent technical decisions.
Requirements
- Experience designing, deploying, and supporting production ML systems.
- Strong theoretical background in NLP and transformer-based approaches.
- Proficiency with modern deep learning frameworks such as PyTorch and common NLP libraries.
- Experience with distributed training of multi-billion-parameter models.
- Strong attention to detail and communication skills.
- Preferred experience with vLLM, DeepSpeed, TensorRT, RLHF/RLAIF, MLOps, CI/CD for ML, Kubernetes, Kubeflow, and scientific NLP publications.
Tech Stack
Categories
AI ResearchML Engineering
About JetBrains
On a mission to make software development a more productive and enjoyable experience. Make it happen. With Code.