Software Engineer - ML Model Training

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 about technical systems and eager to improve them.
• Collaborative, willing to share knowledge and participate in debugging sessions.
• Experience with Canvas/WebGL or visualization libraries like React vis or uPlot.

Skills

• Strong problem-solving skills.
• Ability to work independently and as part of a team.
• Interest in machine learning workflows or a strong interest in the AI/ML ecosystem.

Benefits

• Medical, dental, and vision insurance - 100% paid by CoreWeave.
• Company-paid Life Insurance.
• Voluntary supplemental life insurance.
• Short and long-term disability insurance.
• Flexible Spending Account (FSA) and Health Savings Account (HSA).
• Tuition reimbursement.
• Mental wellness benefits through Spring Health.
• Family-forming support provided by Carrot.
• Paid parental leave.
• Flexible, full-service childcare support with Kinside.
• 401(k) with a generous employer match.
• Competitive salary range of $182,000 to $242,000.
• Discretionary bonus.
• Equity awards.
• A comprehensive benefits program.

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

CoreWeave offers a flexible schedule to support work-life balance.