At Weights & Biases, our mission is to build the best developer tools for machine learning. We're hiring a ML Solutions Architect to help our customers solve difficult, real-world problems and engage in ground-breaking research by using our developer tools in their machine learning pipelines.

In this role you'll be working with the most sophisticated ML teams in the world working on some of the toughest ML problems in computer vision, robotics, natural language processing, and more. Your responsibility will be to make their work easier with our tools. You'll have the opportunity to work with ML teams across multiple industries to uncover their ML needs, improve their ML workflow, explore how W&B fits into their environment, collaborate on projects, and educate them on the best practices of our product.

Solutions Architects are critical to the success of our customers at Weights & Biases. You'll partner with Customer Success, Support, Product and Engineering teams to own the technical onboarding to include installing and deploying our platform both locally or in the cloud, serving as the primary knowledge owner and face to our customers in sometimes complex environments. You'll help drive adoption, understand innovative customer use cases, and serve as the primary problem solver in our customers' machine learning workflows.

Weights & Biases is a series B company with $60 million in funding. Our core users and enterprise customers are growing rapidly and we're facing exciting challenges ahead to keep up with growing demand and to create a platform that is essential to machine learning developers. Our customers range from small startups to enterprise companies including OpenAI, Insitro, Pandora, Toyota Research Institute, Lyft, Blue Cross Blue Shield and Qualcomm.

This is a perfect opportunity for anyone who has machine learning experience, is customer-oriented, and is looking to work with the top ML companies in the world.

Role and Responsibilites

• Work with our customer's operations team to provision W&B services in their cloud environments.
• Debug customer installations when thinks aren't working properly.
• Be an expert in implementing effective, robust, and reproducible machine learning pipelines for engineering teams using Weights & Biases tools
• Effectively articulate best practices for instrumenting machine learning pipelines to our customers as a trusted advisor
• Partner with our customers to uncover their desired outcomes and be the trusted advisor to help them realize the full potential of W&B in solving their problem
• Provide customer training sessions, product demos, and workshops covering best practices & different solutions W&B offers to drive adoption
• Partner with Customer Success Managers to create processes for the post-sales lifecycle (Onboarding/Training, Adoption, Workshops, Demos, etc.)
• Collaborate closely with Support, Product and Engineering teams to influence product roadmap based on customer feedback
• Partner with sales engineering team to ensure there's a smooth transition from POC to when a new customer is onboarded

Requirements

• Familiarity with one or more of the following packages: TensorFlow / Keras, PyTorch / PyTorch Lightning
• Web service / distributed system operations experience. Familiarity with kubernetes, networking and cloud managed services such as MySQL and Object Stores.
• Basic proficiency in Python and eagerness to help customers who are primarily users of Python deep learning frameworks and tools be successful
• Excellent communication and presentation skills, both written and verbal
• Ability to effectively manage multiple conflicting priorities, respond promptly and manage time effectively in a fast-paced, dynamic team environment
• Ability to break down complex problems and resolve them through customer consultation and execution.
• Experience with cloud platforms (AWS, GCP, Azure)
• Strong Linux/Unix command line experience.

Strong plus

• Proficiency with one or more of the following packages: HuggingFace, Fastai, scikit-learn, XGBoost, LightGBM, Ray
• Proficiency with Terraform for infrastructure management
• Experience with hyperparameter optimization solutions
• Experience with data engineering and MLOps tooling
• Experience as an ML educator and/or building and executing customer training sessions, product demos and/or workshops at a SaaS company

Who this role is for

• Passion for machine learning: You understand how exciting the ML space is and how quickly it's growing. You're excited to speak with industry leaders at the forefront of ML research and development
• Outgoing and friendly: You'll love this role if you enjoy connecting with real users every day, helping them adopt the Weights & Biases platform, and answering all of their questions
• Technically Savvy: You enjoy instrumenting models in PyTorch, Keras, and TensorFlow and understand complicated workflows
• Autonomous and adaptable: You understand different companies have different workflows, and you're excited to solve these challenges
• Curious and Driven: You want to understand future customers' ML workflow and prove how Weights & Biases will improve their day-to-day

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."
• A best-in-class product in one of the fastest-growing and largest market segments

Our benefits

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

W&B is free for individuals and academic users. We'd love if you created a project and made a report in W&B on any topic that interests you. You can find inspiration and examples of reports at wandb.ai/gallery.

We are looking for individuals located in the United States to cover the pacific time zone and eastern time zone.

We encourage you to apply even if your experience doesn't perfectly align with the job description. 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 careers@wandb.com.