New

Machine Learning Engineer III, AI Agents

Redwood City, CA, United States

WHAT IS BOX? 

Box (NYSE:BOX) is the leader in Intelligent Content Management. Our platform enables organizations to fuel collaboration, manage the entire content lifecycle, secure critical content, and transform business workflows with enterprise AI. We help companies thrive in the new AI-first era of business. Founded in 2005, Box simplifies work for leading global organizations, including AstraZeneca, JLL, Morgan Stanley, and Nationwide. Box is headquartered in Redwood City, CA, with offices across the United States, Europe, and Asia.

By joining Box, you will have the unique opportunity to continue driving our platform forward. Content powers how we work. It’s the billions of files and information flowing across teams, departments, and key business processes every single day: contracts, invoices, employee records, financials, product specs, marketing assets, and more. Our mission is to bring intelligence to the world of content management and empower our customers to completely transform workflows across their organizations. With the combination of AI and enterprise content, the opportunity has never been greater to transform how the world works together and at Box you will be on the front lines of this massive shift.

 

WHY BOX NEEDS YOU

AI is transforming how enterprises work, and Box is building an enterprise-grade Agents Platform at the core of the Box Content Cloud. Our platform, built on LangGraph, enables teams across Box and our customers to design, deploy, and operate AI agents that handle real-world enterprise workflows—from content understanding and generation to intelligent metadata, automation, and complex, multi-step orchestrations.

As a founding ML Engineer on the AI Agents team, you will build and evaluate the foundational agents that power the Box AI ecosystem, including DeepSearch, DeepResearch, Extract, and Compose. You’ll design techniques for intent detection, ranking, evaluation, retrieval-augmented generation (RAG), and multi-agent orchestration, while also establishing metrics and evaluation frameworks to measure agent quality.

Your work will shape how agents retrieve, reason, and act on enterprise content with high accuracy and trustworthiness. You’ll collaborate closely with platform engineers to build the core components of the Agents Platform that enable these agents to run at scale, while also empowering other Box teams and customers to configure and customize agents for their workflows.

 

WHAT YOU'LL DO 

  • Build, evaluate, and evolve foundational agents such as DeepSearch, DeepResearch, Extract, and Compose.
  • Develop techniques for intent detection, query understanding, ranking, and RAG to improve accuracy and relevance.
  • Define metrics, evaluation pipelines, and benchmarks for agent quality, including precision/recall, factual grounding, and latency trade-offs.
  • Research and implement best practices in retrieval, orchestration, and evaluation of multi-agent workflows.
  • Collaborate with platform engineers to design core components that enable secure, reliable, and scalable deployment of agents.
  • Partner with product teams to translate enterprise use cases into agentic solutions, ensuring measurable improvements in user experience.
  • Contribute to technical discussions, share research insights, and help define the roadmap for Box’s agent ecosystem.
  • Participate in our on-call rotation, available at all times while on-call to help respond to and triage any issues that arise.

WHO YOU ARE

  • You are passionate about building and evaluating AI agents that solve enterprise problems.
  • You enjoy working at the intersection of machine learning and distributed systems, bridging research with production.
  • You’ve designed or evaluated ML systems for search, ranking, RAG, or conversational AI.
  • You like to be an owner and strive to do work you’re proud of—both technically and in your team interactions.
  • You are collaborative, curious, and comfortable mentoring or learning from other engineers and ML practitioners.

Required skills

  • Strong background in machine learning, information retrieval, or natural language processing.
  • Proficiency with at least one programming language such as Python, Java, or Scala.
  • Experience designing, training, and evaluating ML models in production.
  • Familiarity with retrieval systems, ranking models, RAG pipelines, or intent classification.
  • BS degree in Computer Science, Machine Learning, or a related field.
  • 3+ years of industry experience building or evaluating ML-powered systems.

Preferred skills

  • Advanced degree in computer science, machine learning, or related field.
  • Hands-on experience with LangChain, LangGraph, or other agent frameworks.
  • Familiarity with LLMs, embeddings, semantic search, indexing, and relevance optimization.
  • Experience with cloud-based ML platforms such as Vertex AI, AWS Bedrock, or SageMaker.
  • Experience with Kubernetes-based systems for deploying and scaling ML workloads.
  • Research or applied experience in evaluation of generative AI systems (factuality, safety, grounding).

LEARN MORE ABOUT BOX AI

Box lives its values, with community and in-person collaboration being a core part of our culture. Boxers are expected to work from their assigned office a minimum of 3 days per week.Your Recruiter will share more about how we work and company culture during the hiring process.

At Box, we believe unique and diverse experiences benefit our culture, our products, our customers, our company, and our world. We aim to recruit a passionate, high-performing workforce that reflects the world we live in. If you are head-over-heels about this role but unsure if you meet all the requirements, we encourage you to apply!

EQUAL OPPORTUNITY

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, disability, and any other protected ground of discrimination under applicable human rights legislation. Box strives to respect the dignity and ‎‎independence of people with disabilities and is committed to giving them the same ‎‎opportunity to succeed as all other employees. Inclusiveness is core to our culture at Box, and we strive to ensure you get the most from your interview experience.

Box makes reasonable accommodations for applicants with disabilities. If a reasonable accommodation is needed to participate in the job application or interview process, please complete this form. Reasonable accommodations may include scheduling adjustments, document dictation and beyond.

Notice to applicants in Los Angeles:  Box, Inc and its related branches will consider for employment, qualified applicants with criminal histories in a manner consistent with the Los Angeles Fair Chair Ordinance.  The Fair Chance Ordinance is provided here. 

Notice to applicants in San Francisco:  Box, Inc and its related branches will consider for employment, qualified applicants with criminal histories in a manner consistent with the San Francisco Fair Chair Ordinance.  The Fair Chance Ordinance is provided here. 

For details on how we protect your information when you apply, please see our Personnel Privacy Notice. If you are a California-resident, please read our California Applicant & Candidate Privacy Notice here.

Box is committed to fair and equitable compensation practices. Actual base salary (or OTE if commissionable role) is dependent upon factors such as: knowledge, skill level, experience, and work location. This role is also eligible for equity and benefits. For more information on benefits, check out our healthcare benefits and additional Box Benefits + Perks.
 
In accordance with OFCCP compliance, here is the Pay Transparency Provision. 

United States Pay Range

$170,500 - $213,000 USD

Create a Job Alert

Interested in building your career at Box? Get future opportunities sent straight to your email.

Apply for this job

*

indicates a required field

Resume/CV*

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter

Accepted file types: pdf, doc, docx, txt, rtf

Select...
Select...
Select...

US Box Self Identification Questionnaire

 
Here at Box our goal is to fully leverage and engage the unique talents/capabilities of diverse teams who feel like they belong. We provide a safe space to contribute your unique skills, ideas, and insights. A welcoming space is provided to learn, try new things, grow your career, test your thinking, and evolve.  Our goal is to build diverse teams with broad representation that matches our customers and the world at large. We also want to make sure that you feel welcomed and supported, connected, included, and are treated fairly and with respect. We also ask that you do the same for others. 
 
Below is a set of demographic questions and completion is entirely voluntary. Any person seeking employment at Box is considered without regards to race, color, religion, national origin, age, sex, marital status, ancestry, physical or mental disability, veteran status, gender identity, or sexual orientation. If you choose to respond to these questions the information will be used, in aggregate only, to help us track the effectiveness of our recruiting efforts and identify areas of improvement in our process. The information collected, or lack thereof, will not be associated with your specific application and will not in any way be used in the hiring process. 
 
 
Veteran Status:
 
If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:
  • A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.
  • A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.
  • An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.
  • An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Voluntary Self-Identification of Disability
 
Why are you being asked to complete this form?
We are a federal contractor or subcontractor required by law to provide equal employment opportunity to qualified people with disabilities.  We are also required to measure our progress toward having at least 7% of our workforce be individuals with disabilities.  To do this, we must ask applicants and employees if they have a disability or have ever had a disability.  Because a person may become disabled at any time, we ask all of our employees to update their information at least every five years.
 
Identifying yourself as an individual with a disability is voluntary, and we hope that you will choose to do so.  Your answer will be maintained confidentially and not be seen by selecting officials or anyone else involved in making personnel decisions.  Completing the form will not negatively impact you in any way, regardless of whether you have self-identified in the past.  For more information about this form or the equal employment obligations of federal contractors under Section 503 of the Rehabilitation Act, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.   
 
1. To help us measure how well we are doing, we are asking you to tell us if you have a disability or if you ever had a disability. Completing this form is voluntary, but we hope that you will choose to fill it out. If you are applying for a job, any answer you give will be kept private and will not be used against you in any way.
 
If you already work for us, your answer will not be used against you in any way. Because a person may become disabled at any time, we are required to ask all of our employees to update their information every five years. You may voluntarily self-identify as having a disability on this form without fear of any punishment because you did not identify as having a disability earlier.
 
How do I know if I have a disability? 
You are considered to have a disability if you have a physical or mental impairment or medical condition that substantially limits a major life activity, or if you have a history or record of such an impairment or medical condition.  Disabilities include, but are not limited to:
 
  • Autism
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, or HIV/AIDS
  •  Blind or low vision
  • Cancer
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or hard of hearing
  • Depression or anxiety
  • Diabetes
  • Epilepsy
  • Gastrointestinal disorders, for example, Crohn's Disease, or irritable bowel syndrome
  • Intellectual disability
  • Missing limbs or partially missing limbs
  • Nervous system condition for example, migraine headaches, Parkinson’s disease, or Multiple sclerosis (MS)
  • Psychiatric condition, for example, bipolar disorder, schizophrenia, PTSD, or major depression
 
Reasonable Accommodation Notice 
 
Federal law requires employers to provide reasonable accommodation to qualified individuals with disabilities. Please tell us if you require a reasonable accommodation to apply for a job or to perform your job. Examples of reasonable accommodation include making a change to the application process or work procedures, providing documents in an alternate format, using a sign language interpreter, or using specialized equipment.
1Section 503 of the Rehabilitation Act of 1973, as amended. For more information about this form or the equal employment obligations of Federal contractors, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.
PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.
Please review our Personnel Privacy Notice here
 
Select...
Select...
Select...
Select...
Select...
Select...

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Box’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Select...
Select...
Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Select...

Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Select...

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.