Baseten
The ML Application Builder for data scientists.
machine learning • developer tools • software engineering • artificial intelligence
11 - 50
Senior Software Engineer - ML Inference and Performance
February 13
🏡 Remote – New York
Baseten
The ML Application Builder for data scientists.
machine learning • developer tools • software engineering • artificial intelligence
11 - 50
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.
Apply Now
