Applied ML Engineer - Inference

Lambda is the #1 GPU Cloud for ML/AI teams training, fine-tuning and inferencing AI models, where engineers can easily, securely and affordably build, test and deploy AI products at scale. Lambda’s product portfolio includes on-prem GPU systems, hosted GPUs across public & private clouds and managed inference services – servicing government, researchers, startups and Enterprises world-wide.If you'd like to build the world's best deep learning cloud, join us. *Note: This position requires presence in our San Francisco office location 4 days per week; Lambda’s designated work from home day is currently Tuesday.What You’ll DoWork on our Inference service, helping us to develop our large-scale platform for running new, cutting-edge models across tens of thousands of GPUsWork with both existing, open-source inference platforms and engines (such as vLLM, sglang, and Dynamo) and also developing new, Lambda-specific componentsWork with the software and system reliability engineers on our Inference team to take cutting-edge concepts from theory to ultra-reliable production environmentsWork with our Machine Learning team to bring across research concepts and help develop exciting new offerings and products.Adapt to new models and architectures and help keep Lambda at the forefront of this rapidly-changing part of the industryHelp us build a high-throughput, low-latency, all-modes inference system running at globally-distributed scaleYou6+ years experience in software engineering, including at least some experience running public, production systems.3+ years experience with machine learning, especially of running inference against models at scale (experience in other areas, such as training, is also valuable)Deep familiarity with current popular machine learning models and paradigms, including large language models, diffusion models, and mixture-of-experts architectures.Strong Python programming skills, and experience with either Go or Rust.A strong desire to rapidly take concepts from theory into production Nice to HaveHave published papers in relevant machine learning fieldsHave presented talks or posters at machine learning conferencesHands-on experience with running live, streaming inference at scaleSalary Range Information Based on market data and other factors, the annual salary range for this position is $325,000-$490,000. However, a salary higher or lower than this range may be appropriate for a candidate whose qualifications differ meaningfully from those listed in the job description.About LambdaFounded in 2012, ~350 employees (2024) and growing fastWe offer generous cash & equity compensationOur investors include Andra Capital, SGW, Andrej Karpathy, ARK Invest, Fincadia Advisors, G Squared, In-Q-Tel (IQT), KHK & Partners, NVIDIA, Pegatron, Supermicro, Wistron, Wiwynn, US Innovative Technology, Gradient Ventures, Mercato Partners, SVB, 1517, Crescent Cove.We are experiencing extremely high demand for our systems, with quarter over quarter, year over year profitabilityOur research papers have been accepted into top machine learning and graphics conferences, including NeurIPS, ICCV, SIGGRAPH, and TOGHealth, dental, and vision coverage for you and your dependentsWellness and Commuter stipends for select roles401k Plan with 2% company match (USA employees)Flexible Paid Time Off Plan that we all actually useA Final Note:You do not need to match all of the listed expectations to apply for this position. We are committed to building a team with a variety of backgrounds, experiences, and skills.Equal Opportunity EmployerLambda is an Equal Opportunity employer. Applicants are considered without regard to race, color, religion, creed, national origin, age, sex, gender, marital status, sexual orientation and identity, genetic information, veteran status, citizenship, or any other factors prohibited by local, state, or federal law.

Apply Now