Back

AI Solutions Engineer

Worldwide Salaried Open

Baseten powers mission-critical inference for the world's most dynamic AI companies, and they are seeking an AI Solutions Engineer to partner directly with customers to architect, build, and deploy high-scale production AI applications. This role involves hands-on coding, as well as aspects of product management and customer success.

Responsibilities

  • Develop and maintain software systems and product features using one or more general-purpose programming languages in a production-level environment, with a preference for Python due to its relevance in ML projects
  • Drive customer impact by designing, implementing, and deploying Baseten solutions end-to-end (problem framing → evaluation → production deployment → monitoring). This involves working with customers’ engineering teams at every stage of the customer journey including: sales, implementation, and expansion
  • Deliver with velocity: turn vague objectives into clear specs and well-defined PoCs so we can rapidly ship well-tested services and outcomes for our customers
  • Optimize and enhance AI/ML projects, contributing to the continuous improvement of our technical stack. This includes developing features and PRDs with other engineering and product orgs
  • Own products and customer projects end-to-end, functioning as both an engineer, project manager, and product manager, with a focus on user empathy, project specification, and end-to-end execution
  • Navigate ambiguity and exercise good judgment on tradeoffs and tools needed to solve problems, avoiding unnecessary complexity
  • Demonstrate pride, ownership, and accountability for your work, expecting the same from your teammates

Skills

  • Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or related field
  • 1+ years of professional work experience in a fast-paced, high-growth environment
  • Demonstrated experience with one or more general-purpose programming languages in a production-level environment, with a strong preference for Python
  • Familiarity with AI/ML pipelines and the lifecycle of ML model development and deployment
  • Strong communication skills, particularly on complex technical topics
  • Experience in building or optimizing AI/ML projects is highly valued

Benefits

  • Competitive compensation, including meaningful equity.
  • 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
  • Paid parental leave
  • Fertility and family-building stipend through Carrot
  • Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Company Overview

  • Baseten is an AI infrastructure company that integrates machine learning into business operations, production, and processes. It was founded in 2019, and is headquartered in San Francisco, California, USA, with a workforce of 201-500 employees. Its website is https://www.baseten.co.
  • Company H1B Sponsorship

  • Baseten has a track record of offering H1B sponsorships, with 1 in 2026, 6 in 2025, 8 in 2024, 1 in 2023, 1 in 2020. Please note that this does not guarantee sponsorship for this specific role.
  • Apply To This Job

    More jobs

    Software Engineer II, Machine Learning (Feature Platform)

    Worldwide Salaried

    Software Engineer II, Machine Learning (Feature Platform)

    Worldwide Salaried

    ECommerce Requirements Analyst (Hybrid, Birmingham, AL)

    Worldwide Salaried

    U.S. Life Sciences Specialist - 2027

    Worldwide Salaried

    Associate, Client Service, 2026 (CV Drop)

    Worldwide Salaried

    Project Coordinator

    Worldwide Salaried

    Business Analyst - Salesforce

    Worldwide Salaried

    Analyst - PMO Operations Analyst

    Worldwide Salaried

    U.S. Life Sciences Specialist - 2027

    Worldwide Salaried

    Application Analyst - Process Automation (Hybrid)

    Worldwide Salaried

    Linguist/Transcriber - Swahili (Democratic Republic of Congo) - Remote

    Worldwide Salaried

    Experienced Live Chat Customer Service Representative – Delivering Exceptional Apple Customer Experiences Remotely

    Worldwide Salaried

    Member of Technical Staff, ML Compiler and Systems (Bay Area, Remote)

    Worldwide Salaried

    Director of Inside Sales (Remote)

    Worldwide Salaried

    Work from Home - Customer Service Coordinator - arenaflex CX Solutions

    Worldwide Salaried

    Experienced Entry-Level Customer Support Associate – Live Chat (Remote / No Experience) at arenaflex

    Worldwide Salaried

    Product Marketing Manager

    Worldwide Salaried

    Senior Manager - Strategic Sourcing Indirect Spend (Denver Preferred Open to Remote)

    Worldwide Salaried

    Early Childhood Development Technical Lead

    Worldwide Salaried

    Lifecycle Marketing / CRM Consultant (User Flows, Email, Automation & Engagement Systems)

    Worldwide Salaried