Jobs · Engineering · New Jersey

Software Engineer - ML Model Training

Weights & Biases · Livingston, NJ · 6 days ago
Engineering$182k–$242k/yrFull-time

About the role

The successful candidate will join the fullstack W&B History team, focusing on the core visualization engine that powers experiment tracking for over 1,000 organizations and labs building AI across various domains including LLMs, agents, robotics, drug discovery, and more.

Responsibilities

  • Scale charting components to handle millions of data points with smooth interactivity.
  • Implement new features that help AI researchers explore their data and build better models.
  • Collaborate with product designers to refine the user experiences.
  • Work with back-end engineers to optimize data streaming for charts.

Requirements

  • Extensive experience with modern JavaScript/TypeScript and React.
  • Knowledge of web-based visualization libraries.
  • Demonstrated ability to measure and optimize front-end performance, especially for rendering large datasets and maintaining 60fps interactions.

Qualifications

  • Curious and eager to build deep knowledge of technical systems to improve how they function.
  • Collaborative, willing to share knowledge with the team and participate in debugging sessions.
  • Experience with Canvas/WebGL or visualization libraries like React vis or uPlot.

Skills

  • Technical proficiency in modern JavaScript/TypeScript and React.
  • Expertise in web-based visualization libraries.
  • Performance engineering skills, particularly for handling large datasets and maintaining smooth interactivity.

Benefits

  • Base salary range: $182,000 to $242,000.
  • Discretionary bonus.
  • Equity awards.
  • A comprehensive benefits program including medical, dental, and vision insurance, company-paid life insurance, short and long-term disability insurance, flexible spending account, health savings account, tuition reimbursement, mental wellness benefits, employee stock purchase program, parental leave, and childcare support.

Pay

The base salary range for this role is $182,000 to $242,000. The starting salary will be determined based on job-related knowledge, skills, experience, and market location.

Schedule

We are currently in an exciting stage of hyper-growth, building a product central to the AI revolution. We offer a casual work environment and flexible PTO.

Similar jobs