Deep Learning Engineer at Weights & Biases

San Francisco
a year ago

At Weights & Biases, our mission is to build the best developer tools for machine learning. W&B helps ML engineers train high-performing, fair, and reproducible models; visualize and analyze their progress; and share their work effectively with collaborators and the whole field. You would become an early full-time member of a new team, whose primary focus is guiding product development from our core users' perspective. To deliver the most powerful, flexible, and intuitive tools for cutting-edge research and robust applications, we need to understand and synthesize the latest advances in the field of machine learning. As the set of possible workflows and developer needs expands, you will help us test, explore, prototype, and prioritize different features to balance generality with simplicity.In this role, you will work on a diverse, complex, and exciting set of problems at the intersection of engineering, deep learning, and UI/UX design, all in a fast-moving startup environment. Weights & Biases is a series B company with $60 million in funding. Our user base is growing rapidly, and our customers range from small startups to enterprise companies including OpenAI, Insitro, Toyota Research Institute, Lyft, NVIDIA, Github, Salesforce, iRobot, and Blue Cross Blue Shield.We love building powerful tools together, delighting our users, and supporting the inspiring work of machine learning researchers and engineers. If you enjoy working on small engineering teams and thrive in an autonomous environment, we encourage you to apply!

Role and responsibilities

  • Actively use W&B features in a variety of scenarios to provide internal product guidance. Contribute ideas and feedback and help prioritize feature requests and bug fixes.
  • Design and build clear and compelling example projects to guide users and showcase our product to customers: end-to-end ML models with reproducible workflows, illustrative tutorials and how-tos for specific tasks in W&B, general explanatory blog posts and presentations, etc.
  • Listen to and partner with our customers, academic users, and the broader community to understand and help prioritize their workflows in W&B
  • Maintain ontology of ML verticals, techniques, etc. Keep the organization apprised of new developments in applied & theoretical ML
  • Prototype new features for visualization and analysis in cutting-edge deep learning research

Relevant skills and experience

  • Applied machine learning work in industry: training, tuning, debugging, and deploying machine learning models integral to a product or service, in a collaborative team environment
  • Familiarity with a range of ML frameworks and domains (computer vision, natural language processing, reinforcement learning, statistics, etc)—generalization and a high learning rate are more important here than specific architectures or priors
  • Building internal tools and/or giving internal demos and interactive access to your ML models, e.g. sharing scripts, notebooks, or an endpoint to help your team visualize results, understand model performance, or evaluate improvement across versions
  • Caring deeply about the user experience for what you build: thinking through the details, anticipating and testing the edge cases, and considering future applications
  • Writing easy-to-follow code and effective documentation for your projects
  • Ability to clearly communicate your ideas to folks across a range of backgrounds and levels of technical knowledge
  • Strong writing and data visualization skills

Strong plus

  • You have thought deeply about your role in the future of artificial intelligence/technological advancement and want to help make machine learning more accessible, transparent, and collaborative to enable better planetary outcomes
  • Visual design skills and an eye for intuitive UI/UX
  • Your research interests prioritize practical applications, democratization, and social good over popularity, recency, or hype
  • Experience with data pipelines, infrastructure, and ML Ops
  • Experience contributing to architecture and systems design
  • Teaching experience

Our Benefits

  • 100% health insurance coverage for employees, including vision, dental, and family coverage
  • 401k
  • Unlimited vacation time
  • Parental leave
  • Competitive salary
  • Equity
  • Remote flexibility

We encourage you to apply even if your experience doesn't perfectly align with the job description. We value diverse, creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish here. 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 FOR THIS JOB

Your application

Apply directly on company listing page or fill out the form and we will forward it to the company contact.

This information will be shared with the company contact only. Review your information carefully, as the application cannot be edited after submission.

Why are you a great fit for this job?