Post Sales Machine Learning Engineer

*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 Do

• Guide new customers through the technical onboarding process by:

  • Assisting ML researchers in migrating their existing workloads to Lambda’s AI Cloud Platform, ensuring that expected performance is achieved
  • Providing initial troubleshooting for technical issues that arise during the first few days of customers time on Lambda infrastructure

• Collaborate closely with customers to understand their needs and objectives, offer tailored guidance and best practices for deploying models and managing GPU infrastructure

• Demonstrate how to optimize and scale training and inference workloads within Lambda by:

  • Building proof-of-concept demos
  • Creating detailed architecture diagrams

• Create and maintain detailed documentation including technical guides, best practices and troubleshooting resources

• Conduct training sessions and workshops for customers, enabling them to effectively utilize Lambda’s products and services

• Facilitate smooth workload transitions between Lambda’s various products

• Drive customer growth by identifying opportunities to increase product adoption

• Act as a trusted advisor to new customers, ensuring successful integration and optimization of Lambda products

• Provide continuous customer feedback to influence product roadmap and enhancements

• Serve as a link between customers and internal teams

You

• Have experience in machine learning or data science with a deep understanding of model development, and deployment
• Have experience using deep learning frameworks and libraries such as PyTorch, Tensorflow, Deepspeed, etc.
• Have experience with containerization technologies such as Docker and Kubernetes
• Have experience building and optimizing LLM-based applications
• Have experience building end-end ML pipelines on major cloud platforms
• Have experience with Linux systems administration
• Are an excellent communicator, capable of explaining complex, technical concepts to technical and non-technical audiences
• Are customer obsessed, and strive to deliver exceptional experiences to current and future Lambda customers
• Experience as an ML educator and/or building and executing customer training sessions, product demos or workshops

Nice to Have

• Experience using MLOps tools such as RunAI, Weights and Biases, ClearML

• Experience in training large models using distributed systems

  • Selecting parallelism strategies
  • Multi-GPU and Multi-Node training
  • Troubleshooting and configuring NCCL/RDMA
  • Quantization

• Experience with HPC orchestration technologies such as SLURM

• Experience with automation tools like Ansible, Puppet, Salt

Salary Range Information

Based on market data and other factors, the salary range for this position is $144,000 - $210,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.