Senior ML Engineer: LLM Systems & RAG Evaluation

Grid Dynamics; Zug; Vor Ort; CHF 140.000 - 190.000; Vollzeit

Benefits dieser Stelle

Flexible schedule

Medical insurance

Sports

Well-equipped office

Zusammenfassung

Grid Dynamics is seeking a Senior Machine Learning Engineer to lead end-to-end ML projects focusing on LLMs, RAG and agents. You will design evaluation strategies, build benchmarks, and optimize AI systems while collaborating with product and research teams.

The role demands robust Python skills, deep ML fundamentals, and experience with PyTorch, TensorFlow, or JAX. You will shape data pipelines, tooling, and safety modules, translating findings into actionable business insights and scalable

Qualifikationen

• 5+ years of experience in Machine Learning Engineering or related fields.
• Deep understanding of machine learning fundamentals and model evaluation techniques.
• Strong Python skills with experience in modern ML frameworks such as PyTorch, TensorFlow, or JAX.
• Proven experience training, fine-tuning, or adapting large-scale models.
• Hands-on experience working with LLMs beyond simple API integration.
• Ability to evaluate AI systems and translate results into actionable insights.
• Experience building and maintaining ML pipelines and systems.
• Knowledge of RAG architectures, agentic systems, and AI safety concepts.
• Excellent communication skills, both written and verbal.
• Willingness to work up to 9 pm Swiss time.

Aufgaben

• Lead end-to-end machine learning projects from problem definition to deployment.
• Design and implement evaluation methodologies for AI and ML systems.
• Develop datasets, benchmarks, and metrics to measure performance.
• Evaluate and optimize LLM-based systems, including RAG, agents, and safety modules.
• Analyze model behaviors, identify failure modes, and recommend practical improvements.
• Build and maintain ML pipelines, tooling, and evaluation infrastructure.
• Collaborate closely with product, engineering, and research teams to align ML objectives with business goals.
• Prototype rapidly and iterate to solve complex business and product challenges.
• Communicate technical findings, trade-offs, and recommendations to diverse stakeholders.

Tools

PyTorch

TensorFlow

JAX

Python

Jobbeschreibung

Grid Dynamics is seeking a Senior Machine Learning Engineer to lead end-to-end ML projects focusing on LLMs, RAG and agents. You will design evaluation strategies, build benchmarks, and optimize AI systems while collaborating with product and research teams.

The role demands robust Python skills, deep ML fundamentals, and experience with PyTorch, TensorFlow, or JAX. You will shape data pipelines, tooling, and safety modules, translating findings into actionable business insights and scalable