Baseten
Post-Training Research Engineer
San Francisco
Sponsorship not specifiedDetected 121 days ago
KubernetesMachine LearningTensorFlowPyTorchSparkNLPTCP/IPResearch
About the role
- By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production.
- We care about training a wide spectrum of different model architectures with a variety of techniques efficiently and at scale.
- Apply now to embark on a rewarding journey in shaping the future of AI!
Skills
- A deep understanding of modern ML techniques and tools for training transformers
- Advanced experience in a tensor/array computation library like PyTorch, TensorFlow, Jax, or similar
- The experience and knowledge to profile and improve the performance of a distributed GPU program in PyTorch or a similar library
- A willingness to dive into messy problems, work with researchers, derive specifications by asking important questions, and execute
- Familiarity with HPC and distributed computing platforms like Slurm, Ray, Kubernetes, and Dask
- Familiarity with cluster networking technology like Infiniband, RoCE, GPUDirect
- Solid fundamentals in operating systems concepts like processes, files, kernel drivers, containerisation, and networking protocols
- A sense of creativity and willingness to ask difficult questions about our approach, assumptions, and tooling choices
Compensation
- Competitive compensation, including meaningful equity.
Benefits
- Competitive compensation, including meaningful equity.
- 100% coverage of medical, dental, and vision insurance for employee and dependents
- Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
- Paid parental leave
- Fertility and family-building stipend through Carrot
- Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
- If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.
Company info
- At Baseten, we are committed to fostering a diverse and inclusive workplace.
Apply directly at Baseten →Create a free account for alerts like thisView Baseten immigration profile
This listing is sourced directly from Baseten's careers page and normalized into a canonical job model.