Machine Learning Intern

Key Skills

Python Programming, Deep Learning, PyTorch, TensorFlow, Data Modelling

Job Description

The Machine Learning Intern will partner closely with the ProductML team to prototype new machine learning workflows and visualization types in Weights & Biases, provide feedback on product features and showcase their findings via Interactive Reports.

Project proposal requirement: Please submit a 1-page proposal of a project plan for your internship (8-10 weeks full time). Any project applying or advancing concepts in deep learning, machine learning, or data science is welcome. This proposal should include a clear research direction/topic(s), milestones, and strategy outline for your work, including the dataset(s), method(s), relevant existing work/repos, and core research questions you'd like to address/experiments you'd like to run. You should also include a description of how you will use Weights & Biases product features to achieve your goals. We're especially interested in projects focusing on recent research advances, new data visualizations/analysis features which would help our users, and concrete positive applications for the world (e.g. environmental sustainability, addressing algorithmic fairness & bias).

What you’ll achieve:

• Integrate W&B into deep learning project(s) to visualize, explain, and advance their experiments and modeling
• Provide detailed, prioritized internal feedback and file bug reports on W&B features and workflows
• Create compelling interactive Reports to showcase your findings (see examples in our gallery: https://wandb.ai/gallery)
• (optional) Partner with developers as they build and ship features

What we’re looking for:

• Fluent in Python and at least one major deep learning framework (PyTorch, Tensorflow, etc.)
• Actively developing new techniques or applications of deep learning, machine learning, or data science models
• Passionate about learning, sharing your research, and writing for all audiences

Core skills:

• Here are some key characteristics that will help you thrive in this role:
• Outgoing and friendly: Product research requires interfacing with all teams at Weights & Biases. We seek someone who is adept at communicating complicated material with everyone — from scientists and engineers to the general public. We also encourage all of our employees to connect with our users; you'll love this role if you enjoy seeking users to better understand how they use our product.
• Autonomous: We value employees who thrive in a self-directed environment and proactively find ways to improve processes and collaborate with team members or engaged users.
• Curious and driven: Explore machine learning and learn more about the engineering stack and common ML workflows. Solve problems in both fast-paced, short-term sprints and in larger, more long-term projects.
• Organized: A core part of product research at Weights & Biases is delivering feedback in a digestible format for all stakeholders involved — including engineers, product managers, and customers. Your organization skills and time management will be key to running this process well.

Why join us?

• Top-tier machine learning teams rely on our tools for their daily work at companies including OpenAI, Toyota Research Institute, Lyft, Samsung, and Pandora.
• You'll never stop learning. This role gives you first-hand experience talking with leading researchers in the field, understanding their problems, and directly shaping the product direction.
• Our experienced founding team has successfully built and sold ML tools in the past at Figure Eight, and their deep knowledge of our industry, empathy for our users, and skillful management is driving W&B to success.
• Customers genuinely benefit from our tool. Here's a quote from Wojciech Zaremba, Cofounder and Robotics Lead, OpenAI: W&B allows to scale up insights from a single researcher to the entire team, and from a single machine to hundreds of them.

Our Benefits

• Unlimited vacation time
• 100% Medical, Dental, and Vision for employees and Family Coverage
• Remote first culture within office flexibility in San Francisco
• $500 home office budget with a new high-powered laptop
• Truly competitive salary and equity
• 12 weeks of Parental leave
• 401(k)