Software Engineer, SDK
At Weights & Biases, our mission is to build the best developer tools for machine learning. Weights & Biases is a series C company with $250 million in funding and a rapidly growing user base. Our platform is an essential piece of the daily work for machine learning engineers, from academic research institutions like FAIR and UC Berkeley to massive enterprise teams including iRobot, OpenAI, Toyota Research Institute, Samsung, NVIDIA, Salesforce, Blue Cross Blue Shield, Lyft, and more.
The Machine Learning Software Engineer, reporting to the SDK Team Manager, will be responsible for building the software development kit (SDK) which helps our users accelerate their machine learning projects and build better models faster. The role involves exploring the pain points experienced by Machine Learning Engineers who are building state-of-the-art models, designing solutions to address these issues, and implementing efficient and intuitive interfaces which integrate cleanly into the latest Machine Learning frameworks. The role involves working at the intersection of software development, machine learning, and novel product development.
Responsibilities
- Collaborate with senior engineers to design and build a multi-platform / multi-language Machine Learning SDK in a public github repository
- Build out support for advanced machine learning workflows, such as distributed training
- Build, extend, and enhance a highly reliable and performant data ingest SDK
- Design thoughtful solutions to capture user issues in different client setups, and capture the pertinent info for debugging
- Creatively improve on current features, and seek out ways to simplify the API and streamline the user experience
- Improve CLI and SDK interfaces, implement new integrations with popular ML frameworks as they're released
- Keep up with the latest trends in the ML world and leverage existing tools and frameworks whenever necessary
Requirements:
- 4+ years of professional software development experience
- Proficiency in Go, Python or Rust
- Communication and collaboration on cross-functional projects
- Creative problem-solving, and a willingness to dig into details
- Experience contributing to architecture and systems design
- Experience with ML frameworks such as PyTorch or TensorFlow
Our Benefits
- 🏝️ Flexible time off
- 🩺 Medical, Dental, and Vision for employees and Family Coverage
- 🏠 Remote first culture with in-office flexibility in San Francisco
- 💵 Home office budget with a new high-powered laptop
- 🥇 Truly competitive salary and equity
- 🚼 12 weeks of Parental leave (U.S. specific)
- 📈 401(k) (U.S. specific)
- Supplemental benefits may be available depending on your location
- Explore benefits by country
We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out at careers@wandb.com.
#LI-Remote
We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out at careers@wandb.com.
#LI-Remote