Software Engineer (Model Performance Systems)

  • We are looking for Senior Software Engineers to join our team. This is a specialized, high-impact role sitting at the intersection of high-performance computing (HPC) and Large Language Model (LLM) engineering
  • You will not just be building the automated “speedometer and diagnostic” suite for our next-generation AI infrastructure; you will be defining the roadmap, driving key technical decisions, and taking full ownership of the future of this work
  • Benchmarking: Evaluate, run and automate standard LLM quality benchmarks (GSM8K, MMLU) alongside custom performance suites for specific workloads (e.g., long-context window, KV cache reuse, disaggregated serving)
  • DevEx Improvement: Develop and maintain internal GPU-enabled development environments (similar to GitHub Codespaces). You will ensure the team has seamless, high-performance “dev machines” optimized for model experimentation
  • Tool Development: Build and contribute to open-source tools such as InferenceMAX and genai-bench to automate model evaluation, benchmarking and analysis
  • System Profiling: Use profilers like PyTorch Profiler, NVIDIA Nsight Systems and py-spy to collect performance profiles, identify bottlenecks, and debug the compute/networking stack
  • Monitoring & Observability: Develop real-time dashboards and alerts to monitor system health, model startup times, and runtime performance
  • Continuous Integration: Automate performance testing via CI/CD pipelines to catch regressions and build release workflow automation for the model runtimes stack
  • Optimization Automation: Build tools to find the “Pareto frontier”—identifying the absolute best configuration (latency vs. cost vs. quality) for a given model and workload

Benefits

  • Remote-first work environment. The Baseten team is welcome to work from wherever they want; fully remote, in our San Francisco office, or a mix of both. Today, our team (including our founding team) is spread across the United States, Canada, and Armenia. We provide a $1,000 stipend for you to make your home-office comfortable and productive
  • Regular in-person team summits. We get together as a team three times a year to plan, workshop, and most importantly, get to know each other better
  • Unlimited PTO. We ask that everyone take at least 4 weeks of vacation. And we have a company-wide break between Christmas and New Year’s Day
  • Full healthcare coverage. Medical, dental and vision insurance for you and your family
  • Paid parental leave. 16-weeks fully paid parental leave (adoptive and non-birth parents included) and flexibility with schedules while returning to work
  • Company-sponsored 401(k) for you to contribute to
  • Learning and development budget. We encourage you to take classes, attend conferences, and invest in your craft and we’ll cover expenses to make it happen
Back to blog

Other Jobs To Apply

Amazon Delivery Driver

REMOTE - Entry-Level Management Training (Work from Home)No Experience Needed Full Training Provided

Delta Airlines (Data Entry) Work From Home Position Jobs - Apply Now

Film & TV Development & Production Internships: LA - Summer 2026

`Remote Entry-Level Role | Beginner Friendly | Full/Part-Time | Start ASAP

Work From Home Remote Entry Level -Focus Group Position

Fulfillment Center Warehouse Associate

Robotics Warehouse Associate

amazon warehouse associate $15+/ Hour (Sign on Bonus)!

Amazon Warehouse

Sr. Operations Manager

Delivery Driver(06993) - 2753 W Washington

"Fully Remote Benefits Representative | Flexible Schedule | Hiring Immediately

Fulfillment Workplace Health and Safety Specialist Intern 2026 - AZ, CA, CO, NM, NV, OH, OK, TX, UT, WA - Now Hiring

Software Development Engineer III (ADBL189)

Want to Work From Home? Provide Customer Service Support for Leading Cruise Lines - Flexible Hours

DJZ5 - $23.75hr - New Jersey - Amazon DSP Delivery Driver

EHS Specialist

Sales Associate- Redding

Team Member

Remote Travel Specialist (part time)

Remote Data Entry -Time Focus Group Participants (Up To $750/Week)

Data Entry Clerk Work From Home - Part-Time Focus Group Participants (Up To $750/Week)

Remote Healthcare Customer Service Representative

-Urgent Hiring | Entry-Level Remote Role | Flexible Schedule | Apply Today

REMOTE - Entry-Level Management Training (Work from Home)No Experience Needed Full Training Provided

Easy Remote Academic Transcription Jobs No Experience

Immediate Hiring: Virtual Scheduling Assistant-Entry Level

Remote Live Chat Assistant – Entry-Level Customer Support Specialist (No Experience Required | $35/Hour)

Flight Monitor, Amazon - Prime Air

Amazon Prime Delivery Driver $18-$26/hr

Sr. Operations Manager

EHS Specialist

Warehouse Team Member

Warehouse Associate - $18.00/hour

Work From Home - Up to $750/ week

(Part Time) Data Entry - Remote Jobs

Part-Time Doorstep Trash Collection Specialist - Weekends

Front Desk Concierge (Part-Time / Weekend)

EHS Specialist

Area Manager, Early Career (2027) - AR, MS, OK, LA

Warehouse Worker - Central NJ-OCS (collapsed into 763300000)

Amazon Customer Service Representative (Remote)

Remote Customer Service Specialist

Heavy Equipment Operator

Certified Nursing Assistant (CNA) Jobs in Tucson, AZ — $19–$21/hr, Flexible Shifts

Remote Travel Booking Agent (Flexible Schedule)

Pathways Operations Manager Intern - Summer 2027 - Now Hiring

Supervisory Transportation Security Officer

Paid Panel Immediate Start - Remote Part-Time - Customer Service Representative Welcome