Senior Software Engineer - ML Inference and Performance

February 13

Apply Now

Get a free AI resume review

Baseten

WebsiteLinkedInAll Job Openings

The ML Application Builder for data scientists.

machine learning • developer tools • software engineering • artificial intelligence

11 - 50

💰 $8M Seed Round on 2022-04

Description

• Scaling , refining, and fine-tuning the performance of our ML models and its infrastructure • Deep dive into underlying codebases of Pytorch, Transformers, CUDA, and other libraries to debug ML performance issues • Optimize, and scale machine learning models, particularly large language models. • Collaborate with a diverse team to design and implement innovative solutions.

Requirements

• 3 -10 years of experience with one or more general purpose programming languages, including but not limited to Python, Ruby, or Go. • Deep understanding of software engineering principles and a proven track record of developing and deploying AI/ML inference solutions • Strong familiarity with ML libraries, especially PyTorch. • Demonstrated interest and experience in machine learning and large language models. • Deep understanding of GPU Architecture • Familiary with Docker and Kubernetes

Benefits

• Competitive compensation package (Unlimited PTO, 401k, covered healthcare premiums, remote-first culture) • A unique opportunity to be part of a rapidly growing startup in one of the most exciting engineering fields of our era. • An inclusive and supportive work culture that fosters learning and growth. • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.