Back to Jobs

Software Engineer, Models - US

Remote, USA Full-time Posted 2026-06-17
At Weights Biases, our mission is to build the best tools for AI developers. We founded our company on the insight that while there were excellent tools for developers to build better code, there were no similarly great tools to help ML practitioners build better models. Starting with our first experiment tracking product, we have since expanded our solution into a comprehensive AI developer platform for organizations focused on building their own deep learning models and generative AI applications. Weights Biases proudly serves over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.This product-focused Software Engineer will work on our Models team. Thousands of Machine Learning Engineers use our product daily to monitor and understand their ML training runs. To learn more about what you’ll be working on, see an example workspace or try our tutorial (the collaboration notebook lets you try it out with no ML experience necessary to run) - we are proud of how simple the app is to get started using.Our Models team primarily work on features that enhance and centralize the machine learning development workflow for our users. Engineers in this role will be directly working with customers and other engineering teams to build products and customer experiences that will continue to enable the acceleration of machine learning model development.

Responsibilities:

  • Lead new product initiatives that help our customers (ML engineers) collaborate and share information.
  • Collaborate with other developers on complex technical and performance challenges.
  • Find ways to make the developers on the team more productive.
  • Focus on our end users and listen to customer feedback in order to improve our product, either by working directly with customers, or with our team of customer support personnel, designers and PMs.

Requirements:

  • Multiple years in Full-stack engineering, building and maintaining user facing products. Experience with time series data and data visualizations is a plus.
  • Advanced proficiency in modern web technologies such as React, Typescript.
  • Strong software engineering fundamentals and knowledge of at least 1 modern programming language (Java, Go, Python, C++, etc).
  • Experience building high quality products end to end - technical design, implementation, testing, and launch.
  • An ability to navigate ambiguity and reach a clear understanding of problems affecting our users.
  • An ability to build features based on complex requirements stated in user terms, either from users directly, or from our product/customer support organizations.
  • Interest in the AI/ML space, and a desire to continue to learn.
  • Fluent in English, both written and spoken, in order to communicate with our global team and stakeholders.

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 WB to success.
  • Customers genuinely benefit from our tool. Here's a quote from Wojciech Zaremba, Cofounder and Robotics Lead, OpenAI: "WB allows to scale up insights from a single researcher to the entire team, and from a single machine to hundreds of them."

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
  • Supplemental benefits may be available depending on your location

Export Control Compliance

This position requires access to export controlled information. To conform to U.S. Government export regulations applicable to that information, applicant must either be (A) a U.S. person, defined as a (i) U.S. citizen or national, (ii) U.S. lawful permanent resident (green card holder), (iii) refugee under 8 U.S.C. § 1157, or (iv) asylee under 8 U.S.C. § 1158, (B) eligible to access the export controlled information without a required export authorization, or (C) eligible and reasonably likely to obtain the required export authorization from the applicable U.S. government agency. Weights Biases may, for legitimate business reasons, decline to pursue any export licensing process.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 [email protected].

Originally posted on Himalayas

Apply To this Job

Similar Jobs

Senior Technical Account Manager

Remote, USA Full-time

Senior Data Scientist

Remote, USA Full-time

Sales Agent B2B- Deutschsprachig

Remote, USA Full-time

Senior Product Engineer - Ruby

Remote, USA Full-time

TN Licensed 1099 Mental Health Therapist - Remote

Remote, USA Full-time

Consolidation & International Accounting Manager

Remote, USA Full-time

Engineering Manager, Data Pipelines and Integrations

Remote, USA Full-time

Senior Test Automation Engineer

Remote, USA Full-time

ETL Spezialist (m/w/d)

Remote, USA Full-time

Sr iOS Engineer

Remote, USA Full-time

Enterprise Customer Success Manager – Driving Customer Success and Growth at arenaflex

Remote, USA Full-time

Part-Time Remote Data Entry Specialist – Flexible Home-Based Position with arenaflex Healthcare Services

Remote, USA Full-time

Join Today: Customer Service Representative (4am-12pm)

Remote, USA Full-time

Experienced Customer Service Representative – Remote Work-from-Home Opportunity at arenaflex

Remote, USA Full-time

Experienced Customer Service Representative - Telecommute Chat Agent Position at Blithequark - $25-$35/hr with No Prior Experience Required

Remote, USA Full-time

Experienced Medical Customer Service Representative – Delivering Compassionate Care to Patients and Families

Remote, USA Full-time

Netflix Remote Jobs Customer Service (At Work Home)

Remote, USA Full-time

Inpatient Transporter West Side Locations

Remote, USA Full-time

Loan Servicing Officer

Remote, USA Full-time

Experienced Full Stack Customer Service Representative – Remote Work-From-Home Opportunity

Remote, USA Full-time