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.
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
Compensation
$150,000 – $450,000/year + discretionary performance bonus + benefits
The posted salary range represents the company’s good faith estimate of the base compensation for this position upon hire. The actual base compensation offered may vary within this range depending on individual qualifications, including but not limited to relevant skills, experience, education, certifications, and specific business needs. In addition to base salary, total compensation includes an annual discretionary performance bonus and comprehensive benefits.
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
This position is eligible for visa sponsorship.
Similar Jobs
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Build and operate scalable, customer-facing full-stack software for Samsara’s safety products. The role involves developing web, mobile, API, and platform capabilities; integrating LLMs, agentic workflows, and RAG; evaluating AI quality; and improving reliability, scalability, and operational health. The engineer collaborates with Product, Design, ML, and platform teams while owning features from concept through production.
Top Skills:
AWSAzureEinoGoGCPGraphQLJavaLangfuseLlmsMySQLPythonRagReactReact NativeTypescript
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Owns the global procure-to-pay function across procurement operations, accounts payable, supplier onboarding, payments, controls, and financial close. Establishes governance, KPIs, SLAs, SOX compliance, supplier risk standards, payment integrity, and automation initiatives. The role manages vendor relationships, working capital optimization, executive reporting, stakeholder satisfaction, and a distributed team. Responsibilities include hiring, coaching, training, succession planning, process improvement, and scaling P2P operations across multi-entity and multi-currency environments.
Top Skills:
Ai AutomationAribaCoupaErpLean Six SigmaNetSuiteProcure-To-Pay PlatformsZip
eCommerce • Fashion • Retail • Sales • Wearables • Design
Supports luxury retail store operations by assisting customers, operating POS and cash wrap, processing shipments, maintaining stockroom organization, replenishing merchandise, executing visual merchandising updates, and supporting sales floor activities. Requires flexible scheduling, customer service skills, retail experience, attention to detail, physical ability to handle merchandise, and adherence to cash-handling and loss-prevention policies.
Top Skills:
InternetIpadLaptopMobile PosPos SystemsWalkie-Talkie
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
