Staff Software Engineer, Performance- Weights & Biases at Weights & Biases

About the role

Aplyr's Quick Take

This role is for a Staff Software Engineer focused on performance at Weights & Biases, part of CoreWeave. You'll be working on optimizing AI tools and infrastructure, likely involving coding and system design. It's an individual contributor role that requires technical expertise in software engineering and performance optimization.

Good fit

Ideal candidates have several years of experience in software engineering, particularly in performance tuning and optimization. A strong background in AI or machine learning would be beneficial, along with a proactive and collaborative working style.

Worth noting

The company is part of a larger acquisition, which may lead to changes in direction or priorities. The emphasis on AI suggests a fast-paced environment that could be exciting but also demanding.

CoreWeave, the AI Hyperscaler™, acquired Weights & Biases to create the most powerful end-to-end platform to develop, deploy, and iterate AI faster. Since 2017, CoreWeave has operated a growing footprint of data centers covering every region of the US and across Europe, and was ranked as one of the TIME100 most influential companies of 2024. By bringing together CoreWeave’s industry-leading cloud infrastructure with the best-in-class tools AI practitioners know and love from Weights & Biases, we’re setting a new standard for how AI is built, trained, and scaled.

The integration of our teams and technologies is accelerating our shared mission: to empower developers with the tools and infrastructure they need to push the boundaries of what AI can do. From experiment tracking and model optimization to high-performance training clusters, agent building, and inference at scale, we’re combining forces to serve the full AI lifecycle — all in one seamless platform.

Weights & Biases has long been trusted by over 1,500 organizations — including AstraZeneca, Canva, Cohere, OpenAI, Meta, Snowflake, Square,Toyota, and Wayve — to build better models, AI agents and applications. Now, as part of CoreWeave, that impact is amplified across a broader ecosystem of AI innovators, researchers, and enterprises.

As we unite under one vision, we’re looking for bold thinkers and agile builders who are excited to shape the future of AI alongside us. If you're passionate about solving complex problems at the intersection of software, hardware, and AI, there's never been a more exciting time to join our team.

What You’ll Do:

Performance Engineering at Weights & Biases ensures our systems and products scale seamlessly as usage grows. We embed performance thinking into every stage of development—diagnosing issues, designing scalable patterns, and building tools that empower every engineering team to deliver fast, efficient, and reliable experiences. Our mission is to make performance a shared responsibility across the company through data, automation, and collaboration.

About the role:

As a Staff Engineer on the Performance Engineering team, you’ll take technical ownership of complex performance challenges across the W&B platform and infrastructure. You’ll design and lead experiments to improve system scalability and latency, develop standardized benchmarking and load testing frameworks, and drive initiatives to prevent regressions across product surfaces. You’ll collaborate closely with product, infrastructure, and SRE teams, combining deep technical expertise with a holistic understanding of system behavior. Day to day, you’ll blend investigation, architecture, and enablement—helping teams measure, understand, and continuously improve performance.

Who You Are:

8+ years of experience as a software engineer, with at least 2 years focused on performance optimization, systems s

Skills & Tags

python typescript go rust aws ai frontend data

Aplyr's read

Weights & Biases is a hub for AI enthusiasts, offering cutting-edge tools for seamless machine learning development and collaboration.
Synthesized from recent postings & public sources

What's promising

•Strong focus on enhancing productivity for machine learning practitioners.
•Offers comprehensive tools for experiment tracking and model optimization.
•Highly valued by data scientists for improving workflow efficiency.

What to watch

•Limited public information about company culture and work-life balance.
•Potentially high pressure due to rapid AI industry changes.
•Niche focus may limit appeal to broader tech professionals.

Why Weights & Biases

•Specializes in tools specifically designed for machine learning workflows.
•Emphasizes collaboration among data scientists and engineers.
•Provides a platform that integrates seamlessly with popular ML frameworks.

Aplyr’s read is generated by AI from public sources. Was it useful?

About Weights & Biases

Weights & Biases

wandb.ai

View company

Weights & Biases is a leading platform for machine learning practitioners, providing tools for experiment tracking, model optimization, and collaboration. Their solutions empower data scientists and engineers to streamline their workflows and improve productivity in developing AI models.