Senior Software Engineer, ML Model Training UI - Weights & Biases

About The Role

As a Front End Engineer working on Data Visualization, your primary focus will be the architecture and performance of line plots within the W&B UI. You will scale charting components to handle millions of data points with smooth interactivity. You will implement new features that help AI researchers at the world's top labs explore their data and build better models. You will collaborate with product designers to refine user experiences and with back-end engineers on your team to optimize data streaming for charts.

What You'll Do

You will join the fullstack Weights & Biases History team, specifically focusing on the core visualization engine that powers experiment tracking for over 1,000 organizations and labs building AI across LLMs, agents, robotics, drug discovery, and more. Our team is dedicated to building highly performant, interactive, and intuitive tools that help ML practitioners visualize complex training metrics in real-time.

Who You Are

• Technical Proficiency: Extensive experience with modern JavaScript/TypeScript and React.
• Visualization Expertise: Knowledge of web-based visualization libraries.
• Performance Engineering: Demonstrated ability to measure and optimize front-end performance, specifically for rendering large datasets and maintaining 60fps interactions.
• Curious: Work isn't about moving tickets across a board, it's about building deep knowledge of technical systems to improve how they function.
• Collaborative: You like to share what you've learned with others on the team, and you'd enjoy hopping in huddles with others to debug a customer issue.

Qualifications

• Preferred Background in Machine Learning workflows or a strong interest in the AI/ML ecosystem.
• Experience with Canvas/WebGL or visualization libraries like React vis or uPlot.

Benefits

• Medical, dental, and vision insurance - 100% paid for by CoreWeave
• Company-paid Life Insurance
• Voluntary supplemental life insurance
• Short and long-term disability insurance
• Flexible Spending Account
• Health Savings Account
• Tuition Reimbursement
• Ability to Participate in Employee Stock Purchase Program (ESPP)
• 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
• Flexible PTO
• Catered lunch each day in our office and data center locations
• A casual work environment
• A work culture focused on innovative disruption

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. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).