CrawlJobs Logo

Staff ML Engineer - Applied AI

India, Bangalore · Job Posted April 11, 2026
Apply Position
Job Link Share

Job Description

Applied AI is a horizontal AI team at Uber partnering with product and platform teams across the company to deliver cutting-edge machine learning solutions for core business problems. We specialize in areas like Generative AI, Computer Vision, Personalization, and the ML infrastructure needed to scale these systems in production. We’re looking for a Staff Engineer with deep experience in classical machine learning, generative AI, and ML systems design to help build and scale high-impact AI solutions. In this role, you’ll lead technically complex projects, influence the architecture of ML systems, and collaborate cross-functionally to drive innovation and impact across multiple Uber surfaces. This is a great opportunity for a strong technical leader who thrives in a fast-paced, product-driven environment and wants to be at the forefront of applied AI at scale.

Job Responsibility

  • Design and implement ML-driven systems that power core Uber experiences, with a focus on scalability, reliability, and performance
  • Lead the technical execution of key projects involving classical ML, deep learning, and generative AI technologies (e.g., LLMs, multimodal models)
  • Collaborate closely with product, data science, and infrastructure teams to develop AI solutions from ideation through production deployment
  • Contribute to and influence the technical direction for Applied AI, particularly around system design, model architecture, and infrastructure decisions
  • Champion engineering best practices in ML development — including experimentation workflows, model versioning, evaluation, monitoring, and responsible AI
  • Provide mentorship to engineers on the team and across partner orgs to help raise the technical bar

Requirements

  • 10+ years of industry experience in machine learning or software engineering, with a proven record of delivering ML solutions to production
  • Strong knowledge of machine learning, deep learning, and exposure to generative AI techniques (e.g., transformers, LLMs, diffusion)
  • Experience designing and scaling ML systems or platforms, including training pipelines, serving infrastructure, and model lifecycle tooling
  • Fluency in ML frameworks (e.g., PyTorch, TensorFlow, JAX) and development in Python and/or scalable backend languages (e.g., Java, Go)
  • Excellent collaboration and communication skills with the ability to work across teams and functions

Nice to have

  • PhD in Computer Science, Machine Learning, or a related field
  • Hands-on experience integrating LLMs or generative models into product experiences (e.g., summarization, personalization, automation)
  • Familiarity with MLOps, experimentation frameworks, or ML observability tools
  • Track record of technical leadership in multi-disciplinary projects involving engineering, data science, and product

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Similar Jobs for

Staff ML Engineer - Applied AI

8 matching positions

Staff Applied AI Engineer

A rapidly growing venture-backed startup is on a mission to transform how people...
Location
Location
United States
Salary
Salary:
Not provided
weareorbis.com Logo
Orbis Consultants
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • STEM degree from a top school (bonus points for Masters or PhD)
  • 5+ years of hands-on experience applying AI/ML techniques to real-world, user-facing products
  • Strong software engineering background with a track record of shipping at scale
  • Experience tuning and deploying LLMs
  • Comfortable moving quickly, testing new approaches, and balancing experimentation with reliability
Job Responsibility
Job Responsibility
  • Build and refine AI-driven product features that feel intuitive and responsive
  • Apply data from real users to personalise interactions and improve predictive performance
  • Experiment with model fine-tuning, retrieval methods, and feedback loops to maximise accuracy
  • Develop specialised AI-powered workflows that simplify complex technical tasks
  • Collaborate with engineers and product leaders to ship enhancements rapidly and iteratively
What we offer
What we offer
  • meaningful equity on offer
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As a Staff AI Engineer you will get to play with petabyte data gathered from a m...
Location
Location
Salary
Salary:
Not provided
balbix.com Logo
Balbix
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Ph.D./M.S. in Computer Science or Electrical Engineering with hands-on software engineering experience
  • 5+ years of experience in the field of Machine Learning and programming in Python
  • Expertise in programming concepts and building large scale systems
  • Knowledge of state-of-the-art algorithms combined with expertise in statistical analysis and modeling
  • Robust understanding of NLP, Probabilistic Graphical Models, Deep Learning with graphs structures, model explainability, etc.
  • Foundational knowledge of probability, statistics and linear algebra
Job Responsibility
Job Responsibility
  • Design and develop an ensemble of classical and deep learning algorithms for modeling complex interactions between people, software, infrastructure and policies in an enterprise environment
  • Design and implement algorithms for statistical modeling of enterprise cybersecurity risk
  • Apply data-mining, AI and graph analysis techniques to address a variety of problems including modeling, relevance and recommendation
  • Build production quality solutions that balance complexity and performance
  • Participate in the engineering life-cycle at Balbix, including designing high quality ML infrastructure and data pipelines, writing production code, conducting code reviews and working alongside our infrastructure and reliability teams
  • Drive the architecture and the usage of open source software library for numerical computation such as TensorFlow, PyTorch, and ScikitLearn
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer

Machine Learning Engineers at Rocket Money further our mission by building produ...
Location
Location
United States , San Francisco; Washington, D.C.; New York City; Silver Spring; Miami; Denver
Salary
Salary:
210000.00 - 260000.00 USD / Year
truebill.com Logo
Truebill
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 8+ years of professional experience in machine learning engineering or data science roles
  • Proven track record of designing and implementing ML systems at consumer tech scale and speed
  • Extensive hands-on experience integrating ML and AI methods into production workflows, including creating evaluation tooling and effective user feedback mechanisms
  • Experience with prompt engineering and management, creating robust systems for testing and optimizing LLM-based applications
  • Expert-level proficiency in Python, SQL, and at least a handful of common ML frameworks
  • Understanding of ML methods at a fundamental level
  • Master at taking ambiguous problems, creating clarity, and breaking down work into manageable chunks for implementation
  • Owned the development, launch, and maintenance for several scaled ML/AI powered product experiences
  • Understand basic software engineering and computer science fundamentals and have applied them at consumer grade scale to build ML powered products in production environments
  • Technical leader who can identify both emergent technical opportunities and gaps relative to best practice
Job Responsibility
Job Responsibility
  • Lead the architecture and development of complex AI and ML powered features across Rocket Money's product suite
  • Design, implement, and maintain robust evaluation frameworks
  • Develop novel new product experiences
  • Own end to end development and implementation of ML and AI product features in collaboration with cross-functional product development teams
  • Provide technical mentorship
What we offer
What we offer
  • Health, Dental & Vision Plans
  • Competitive Pay
  • 401k Matching
  • Unlimited PTO
  • Lunch daily (in-office only)
  • Snacks & Coffee (in-office only)
  • Commuter benefits (in-office only)
  • Bonus
  • Fulltime
Read More
Arrow Right

Senior Staff Machine Learning Engineer

Help design our AI platform and develop our next generation of machine learning ...
Location
Location
United States , San Francisco
Salary
Salary:
216500.00 - 324500.00 USD / Year
gofundme.com Logo
GoFundMe
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 9+ years of hands-on experience in machine learning engineering, AI development, software engineering, or related fields
  • Experience emphasizing secure, large-scale, distributed system design, AI/ML pipeline development, and implementation
  • Extensive experience designing, developing, and operating scalable backend systems
  • Experience applying software engineering best practices such as domain-driven design, event-driven architectures, and microservices
  • Deep expertise in agentic workflows, AI evaluation solutions, prompt management, and secure AI development and testing practices
  • Strong knowledge of relational and document-based databases, data storage paradigms, and efficient RESTful API design
  • Experience establishing robust CI/CD pipelines, automated testing (unit and integration), and deployment practices
  • Strong leadership skills, including effective planning and management of complex projects, mentoring of team members, and fostering a collaborative, high-performing engineering culture
  • Excellent communicator, able to articulate complex technical concepts clearly to both technical and non-technical stakeholders
  • Bachelor's degree in Computer Science, Software Engineering, or a related technical field (preferred)
Job Responsibility
Job Responsibility
  • Design and implement AI platforms to enable scalable and secure access to LLMs from multiple model providers for diverse use cases
  • Design and implement agentic workflows, agentic tool ecosystems, and LLM prompt management solutions
  • Design, build, and optimize scalable model training, fine tuning, and inference pipelines, ensuring robust integration with production systems
  • Influence technical strategy and approach to developing embedding stores, vector databases, and other reusable assets
  • Lead initiatives to streamline ML and AI workflows, improve operational efficiency, and establish standardized procedures to achieve consistent, high-quality results across our AI systems
  • Design and develop backend services and RESTful APIs using Python and FastAPI, integrating seamlessly with ML pipelines and services
  • Take operational responsibility for team-owned services, including performance monitoring, optimization, troubleshooting, and participation in an on-call rotation
  • Collaborate with both technical and non-technical colleagues, including data and applied scientists, software engineers, product managers, and business stakeholders, to deliver reliable and scalable ML-driven products
  • Coach and mentor fellow ML engineers, promoting a culture of collaboration, continuous improvement, and engineering excellence within the team
  • Employ a diverse set of tools and platforms including Python, AWS, Databricks, Docker, Kubernetes, FastAPI, Terraform, Snowflake, Coralogix, and GitHub to build, deploy, and maintain scalable, highly available machine learning infrastructure
What we offer
What we offer
  • Competitive pay
  • Comprehensive healthcare benefits
  • Financial assistance for things like hybrid work, family planning
  • Generous parental leave
  • Flexible time-off policies
  • Mental health and wellness resources
  • Learning, development, and recognition programs
  • Fulltime
Read More
Arrow Right

Staff / Principal Machine Learning Engineer

Our intelligent runtime must seamlessly connect to foundational models - whether...
Location
Location
United States , Mountain View
Salary
Salary:
240000.00 - 385000.00 USD / Year
inworld.ai Logo
Inworld AI
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • A PhD in a relevant technical field, or a BA/BS degree with equivalent research and/or engineering experience
  • 5+ years of combined experience in software development (Python, C++) and applied ML engineering
  • Demonstrated experience applying or researching ML in domains such as natural language processing, speech processing, and/or action planning
  • Strong foundation in data structures, algorithms, and neural network architectures
  • Proficiency with ML frameworks such as PyTorch
Job Responsibility
Job Responsibility
  • Experiment with and implement cutting-edge ML models and techniques to advance our core AI capabilities
  • Train, evaluate, and optimize production-scale models and systems, focusing on quality, latency, cost, and on-device constraints
  • Collaborate with product and backend teams to translate novel ideas and research findings into robust, production-ready solutions
What we offer
What we offer
  • bonus
  • equity
  • benefits
  • relocation assistance
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer

We are seeking a Staff Machine Learning Engineer to join our Foundation AI team....
Location
Location
United States , Boston
Salary
Salary:
170000.00 - 230000.00 USD / Year
whoop.com Logo
Whoop
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Advanced degree (Master’s or Ph.D.) in Computer Science, Machine Learning, Electrical Engineering, or a related field, or equivalent professional experience
  • 7+ years of experience in applied ML, AI research, or large-scale modeling, with a track record of delivering production systems
  • Expertise in modern deep learning (e.g., transformers, state space models) and multimodal model training
  • Proficiency in Python and deep learning frameworks (e.g., PyTorch, TensorFlow)
  • Experience building and scaling large datasets and training large models in distributed compute environments
  • Strong applied experience with representation learning, self-supervised methods, and fine-tuning for downstream applications
  • Familiarity with MLOps best practices including model versioning, evaluation, CI/CD for ML, and cloud-based compute
  • Excellent communication skills and ability to collaborate cross-functionally with engineers, researchers, and product teams
  • Passion for WHOOP’s mission to improve human performance and extend healthspan through science and technology
Job Responsibility
Job Responsibility
  • Design, train, and optimize large-scale multimodal foundation models that integrate wearable sensor data, text, biomarkers, and behavioral data
  • Conduct applied research in self-supervised learning, representation learning, and downstream task fine tuning to advance WHOOP’s core model capabilities
  • Develop scalable, distributed training pipelines for large models on high-performance compute environments
  • Collaborate with MLOps, data engineering, and software engineering teams to operationalize models for production deployment, ensuring robustness, reproducibility, and observability
  • Partner with product and research teams to translate foundation model capabilities into downstream features that deliver meaningful member value
  • Contribute to the technical roadmap and architectural direction for foundation model development at WHOOP
  • Serve as a technical mentor for other data scientists, sharing best practices in deep learning, large-scale training, and multimodal data integration
  • Ensure models adhere to WHOOP’s standards for ethical, transparent, and privacy-preserving AI
What we offer
What we offer
  • competitive base salaries
  • meaningful equity
  • benefits
  • generous equity package
  • Fulltime
Read More
Arrow Right

Sr Staff ML Engineer - Applied AI

We are building AI-native discovery experiences across Mobility and Delivery. Se...
Location
Location
United States , San Francisco
Salary
Salary:
267000.00 - 297000.00 USD / Year
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Masters degree or Ph.D in Computer Science, Engineering, Mathematics
  • 12+ years of ML experience, including significant work on large-scale deep learning systems
  • Demonstrated ownership of high-impact ML systems in search, recommendations, or conversational AI
  • Deep expertise in transformers, retrieval systems, ranking, and embedding architectures
  • Strong experience with PyTorch and distributed training
  • Proven ability to set the technical strategy for a large organization and influence product roadmaps at the executive level
  • Strong product intuition and ability to connect model improvements to business outcomes
Job Responsibility
Job Responsibility
  • Define and champion the multi-year technical vision and architecture for foundation models across Search, Recommendations, and Conversational AI
  • Set the architectural standard and drive system design for critical, high-leverage ML platforms across Mobility and Delivery
  • Lead cross-team initiatives spanning Retrieval, Ranking, Personalization, and LLM-powered assistants, resolving complex technical trade-offs across organizational boundaries
  • Define long-term investment areas (build vs fine-tune vs partner models) with clear business rationale and long-term viability
  • Provide principal-level technical leadership, mentoring Staff and Senior Staff engineers, and setting the bar for technical excellence across the entire AI organization
What we offer
What we offer
  • Eligible to participate in Uber's bonus program
  • May be offered an equity award & other types of comp
  • All full-time employees are eligible to participate in a 401(k) plan
  • Eligible for various benefits
  • Fulltime
Read More
Arrow Right

Sr. Distinguished AI Engineer (Agentic AI Platform)

At Capital One, we are creating responsible and reliable AI systems, changing ba...
Location
Location
United States , San Jose, California; San Francisco, California
Salary
Salary:
343400.00 - 392000.00 USD / Year
capitalone.com Logo
Capital One
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree in Computer Science, Engineering, or AI plus at least 10 years of experience developing AI and ML algorithms or technologies, or Master's degree plus at least 8 years of experience developing AI and ML algorithms or technologies
  • At least 10 years of experience programming with Python, Go, Scala, or Java
  • 9 years of experience deploying scalable and responsible AI solutions on cloud platforms
  • 2+ years of experience supporting Agentic Frameworks
  • 2+ years of experience with LLMOps
  • 8+ years of experience designing mission-critical machine learning platforms
  • 2+ years of experience architecting, designing, developing, integrating, delivering, and supporting complex AI systems
  • Demonstrated ability to lead and mentor multiple engineering teams and influence cross-functional stakeholders up to the VP level
  • Experience developing AI and ML algorithms or technologies using Python, C++, C#, Java, or Golang
  • Master's degree in Computer Science, Computer Engineering, or relevant technical field
Job Responsibility
Job Responsibility
  • Partner with a cross-functional team of engineers, research scientists, technical program managers, and product managers to deliver AI-powered products
  • Contribute to the north star platform architecture, continuously publishing and refining living diagrams and canonical APIs
  • Standardizing and automating agentic workflows
  • Contribute to crafting an end to end GenAI SDK, CLI and starter kits
  • Help bring together a vision of central guardrail services
  • Collaborate with cross organization architects to drive end to end performance
  • Accelerate innovation by incubating proof of concepts and driving RFCs
  • Own central Helm charts, operators and CRDs that auto scale agents to hit tenant SLAs
  • Coach and evangelize - hosting architecture office hours, mentoring Staff, Principal and Senior engineers, authoring technical design documents and blogs and representing Capital One at Tier1 AI conferences
What we offer
What we offer
  • Performance based incentive compensation, which may include cash bonus(es) and/or long term incentives (LTI)
  • comprehensive, competitive, and inclusive set of health, financial and other benefits
  • Fulltime
Read More
Arrow Right