About Scale AI
At Scale AI, our mission is to accelerate the development of AI applications. This role is at the intersection of cutting-edge AI research and practical application, with a focus on studying the data types essential for building state-of-the-art agents.
Responsibilities
- Explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation.
- Contribute to impactful research publications on agents, collaborate with customer researchers, and work alongside the engineering team to translate these advancements into real-world, scalable solutions.
Requirements
- Practical experience working with LLMs, with proficiency in frameworks like Pytorch, Jax, or Tensorflow.
- A track record of published research in top ML venues (e.g., ACL, EMNLP, NAACL, NeurIPS, ICML, ICLR, COLM, etc.).
- At least three years of experience addressing sophisticated ML problems, either in a research setting or product development.
Nice to Have
- Hands-on experience with open source LLM fine-tuning or involvement in bespoke LLM fine-tuning projects using Pytorch/Jax.
- Hands-on experience and publications in building applications and evaluations related to AI agents such as tool-use, text2SQL, browser agents, coding agents and GUI agents.
- Hands-on experience with agent frameworks such as OpenHands, Swarm, LangGraph, etc.
- Familiarity with agentic reasoning methods such as STaR and PLANSEARCH
- Experience working with cloud technology stack (eg. AWS or GCP) and developing machine learning models in a cloud environment.
Benefits
- Comprehensive health, dental and vision coverage
- Retirement benefits
- A learning and development stipend
- Generous PTO
- Commuter stipend
Salary Range
$259,200-$324,000 USD
XML job scraping automation by YubHub
