The Role
Research and engineer agentic language models: develop algorithms, datasets, evaluations, and scalable RL and distributed training/serving infrastructure; publish papers and contribute open-source software.
Summary Generated by Built In
About the Institute of Foundation Models
We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy.
As part of our team, you’ll have the opportunity to work on the core of cutting-edge foundation model training, alongside world-class researchers, data scientists, and engineers, tackling the most fundamental and impactful challenges in AI development. You will participate in the development of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills will be instrumental in establishing MBZUAI as a global hub for high-performance computing in deep learning, driving impactful discoveries that inspire the next generation of AI pioneers.
Position Summary
As a member of the Agents team, you will tackle research and engineering challenges to train advanced agentic language models that are adept at using reasoning and tool use to complete tasks on a computer. These challenges range from building high-quality training datasets and robust evaluations to innovative reinforcement learning infrastructure, algorithms and environments. The team aims to recruit candidates across the research-engineering spectrum, from algorithmic and empirical research to large-scale systems and optimization, reflecting the breadth of interesting problems we aim to solve. We value depth in any subset of these areas; there is no single “right” background.
Key Responsibilities
· Propose, prototype, and scale algorithms/systems to support agentic learning
· Develop new paradigms for agentic behavior that go beyond current practices
· Contribute to technical reports, research publications, and open-source software
· Collaborate with other teams to produce state-of-the-art foundation models
Academic Qualifications
• BS, MS, or PhD degree (or equivalent experience) in Computer Science, Machine Learning, or related fields
Professional Experience
Minimum
· 2 years of experience working in one of the following or related areas: LLM training/fine-tuning, evaluations, reinforcement learning, LLM inference, distributed machine learning systems
· Python and PyTorch development experience
· Experience in using LLM agents for your personal or professional use
· Experience in designing and implementing algorithms from scratch (for algorithms focus)
· Experience in dataset curation and generation (for data focus)
· Experience in building or contributing to training/serving infrastructure (for systems focus)
· Experience in designing and deep diving into evaluations (for evals focus)
Preferred Skills
· Expertise in one or more of PyTorch, Ray, Triton, CUDA C++
· Strong knowledge of literature on RL, LLM reasoning, and tool use
· Experience in training or using LLM agents for SWE tasks
· Deep understanding of reinforcement learning principles
· Experience implementing distributed learning algorithms
· Contributions to published research and/or open-source ML software
Visa Sponsorship
This position is eligible for visa sponsorship.
Benefits Include
*Comprehensive medical, dental, and vision benefits
*Bonus
*401K Plan
*Generous paid time off, sick leave and holidays
*Paid Parental Leave
*Employee Assistance Program
*Life insurance and disability
Skills Required
- BS, MS, or PhD in Computer Science, Machine Learning, or related field (or equivalent experience)
- 2+ years experience in LLM training/fine-tuning, evaluations, reinforcement learning, LLM inference, or distributed machine learning systems
- Python development experience
- PyTorch development experience
- Experience using LLM agents (personal or professional)
- Experience designing and implementing algorithms from scratch (algorithms focus)
- Experience in dataset curation and generation (data focus)
- Experience building or contributing to training/serving infrastructure (systems focus)
- Experience designing and deep diving into evaluations (evals focus)
- Expertise in PyTorch, Ray, Triton, CUDA C++
- Strong knowledge of RL, LLM reasoning, and tool-use literature
- Experience training or using LLM agents for software engineering tasks
- Deep understanding of reinforcement learning principles
- Experience implementing distributed learning algorithms
- Contributions to published research and/or open-source ML software
Am I A Good Fit?
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.
Success! Refresh the page to see how your skills align with this role.
The Company
What We Do
First a passion, then an idea transformed into success – when it comes to pioneering automation and digitalisation technology, the ifm group is the ideal partner. Since its foundation in 1969, ifm has developed, produced and sold sensors, controllers, software and systems for industrial automation and for SAP-based solutions for supply chain management and shop floor integration worldwide. As one of the pioneers of Industry 4.0, ifm develops and implements consistent solutions to digitalise the entire value chain “from sensor to ERP”. Today, the second-generation family-run ifm group has more than 8,750 employees and is one of the worldwide market leaders. The group combines the internationality and innovative strength of a growing group of companies with the flexibility and close customer contact of a medium-sized company.
