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
eCommerce • Healthtech • Kids + Family • Retail • Social Media
Own the strategy, architecture, development, deployment, monitoring, and continuous improvement of production personalization and recommendation systems. Build custom embeddings, ranking models, and shared representations across feeds, search, and recommendations. Lead ambiguous problems from concept to measurable customer impact, establish AI evaluation standards, partner with product and data teams, make cross-team technical decisions, and coach senior engineers.
Top Skills:
AirflowAWSAws SagemakerBedrock AgentcoreClaude CodeCoderabbitDatadogDbtDeep LearningDevinGithub ActionsHexIncident.IoIterableKotlinKubernetesLangchainLaunchdarklyLinearMatrix FactorizationMaximMlflowMySQLPackwerkPandasPythonPyTorchRuby on RailsRankingReactRetrievalScikit-LearnShopifySidekiqSigmaSnowflakeSwiftTerraformTypescriptWardenWarpstreamWeaviateXgboost
HR Tech • Information Technology • Professional Services • Sales • Software
Manage approximately 200 existing SMB accounts, driving upsells, cross-sells, renewals, and expansion revenue. Own the full commercial sales cycle from opportunity identification through negotiation and contract execution. Build consultative customer relationships, uncover whitespace, forecast pipeline, and use structured sales methodologies. Partner with Customer Success Managers and Sales Engineers while maintaining accurate CRM records and staying current on products, customer trends, and sales tools.
Top Skills:
Gong EngageLinkedin Sales NavigatorSalesforce
HR Tech • Information Technology • Professional Services • Sales • Software
Respond to inbound inquiries, qualify prospects, educate buyers about HiBob’s HRIS platform, schedule discovery meetings for Account Executives, maintain CRM records, follow up with leads, and collaborate with Marketing and Sales to improve pipeline quality and customer experience.
Top Skills:
Crm PlatformsHrisHubspotLive Chat PlatformsSaaSSales Engagement ToolsSalesforce
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
