Research Engineer (Agentic Models) Amsterdam, Netherlands
Core
Design, implement, and maintain SFT and RL post-training pipelines for multi-step coding agents that understand codebases, plan changes, and call tools within JetBrains IDEs.
Role type
Senior Research Engineer (Agentic Models)
Builds
Multi-step coding agents for JetBrains IDEs
Domain
AI/ML, Developer Tools, Large Language Models
Deliverable
production ML models
Required skills
LLM post-training (SFT/RL), PyTorch, distributed GPU training, data pipeline design, evaluation framework design, Python
Preferred skills
ML orchestrators (Kubeflow/Dagster/Airflow), large-scale data processing (MapReduce), experiment tracking (Weights & Biases/MLflow), inference optimization
Technologies
PyTorch, Megatron, NeMo, verl, Kubernetes, SLURM, Weights & Biases, MLflow, Langfuse
Responsibilities
Design and maintain SFT and RL post-training pipelines; Train and adapt LLMs for agent workflows; Build evaluation and simulation environments for coding agents; Design evaluation frameworks and metrics; Analyze training results to improve model architectures; Work with large-scale distributed GPU clusters
Seniority
Senior, hands-on IC
