Logo
Weights & Biases

Staff Software Engineer, Performance- Weights & Biases

Weights & Biases, Sunnyvale, California, United States, 94087

Save Job

Staff Software Engineer, Performance – Weights & Biases 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.

Performance Engineering At Weights & Biases, performance engineering 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 scalability, or observability.

Proficiency in at least one of Go, Python, or TypeScript, with strong systems‑level debugging and profiling skills.

Deep understanding of distributed system concepts (caching, queuing, concurrency, resource contention, etc.) and their performance implications.

Experience designing and implementing load‑testing, benchmarking, or performance‑regression frameworks (e.g., Locust, Gatling, k6, custom solutions).

Familiarity with metrics and tracing systems (Prometheus, Grafana, OpenTelemetry, Datadog, etc.) and performance data analysis.

Experience integrating performance measurement into CI/CD pipelines.

Ability to design experiments, analyze large datasets, and draw actionable conclusions.

Proven track record of driving cross‑functional technical initiatives from design through adoption.

Preferred

Prior experience in performance engineering or SRE at scale (e.g., supporting multi‑region, high‑throughput systems).

Familiarity with frontend performance monitoring and web vitals.

Experience building developer‑facing tools or frameworks.

Background in data systems, observability, or ML infrastructure.

Experience contributing to open‑source performance or benchmarking tools.

Wondering if you’re a good fit?

You love to turn messy, complex performance problems into clear, data‑backed stories that lead to concrete action.

You’re curious about how systems behave under real‑world load and enjoy building experiments that reveal what’s really happening under the hood.

You’re an expert in transforming deep technical insights into tools and processes that empower other engineers to build faster, more reliable systems.

You find satisfaction in long‑term, cross‑team collaboration — helping others succeed by embedding performance thinking into their workflows.

You get energy from bridging theory and practice: designing scalable architectures and writing the code that proves them out.

You care deeply about measurement, iteration, and continuous improvement — not just fixing bottlenecks, but preventing them from ever recurring.

Why Us We work hard, have fun, and move fast! We’re in an exciting stage of hyper‑growth that you will not want to miss out on. We’re not afraid of a little chaos, and we’re constantly learning. Our team cares deeply about how we build our product and how we work together, which is represented through our core values: - Be Curious at Your Core - Act Like an Owner - Empower Employees - Deliver Best‑in‑Class Client Experiences - Achieve More Together

Benefits The base salary range for this role is $188,000 to $275,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).

In addition to a competitive salary, we offer a variety of benefits to support your needs, including: - 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 - Eligibility 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

Our Workplace While we prioritize a hybrid work environment, remote work may be considered for candidates located more than 30 miles from an office, based on role requirements for specialized skill sets. New hires will be invited to attend onboarding at one of our hubs within their first month. Teams also gather quarterly to support collaboration.

Equal Opportunity CoreWeave is an equal opportunity employer, committed to fostering an inclusive and supportive workplace. All qualified applicants and candidates will receive consideration for employment without regard to race, color, religion, sex, disability, age, sexual orientation, gender identity, national origin, veteran status, or genetic information. As part of this commitment and consistent with the Americans with Disabilities Act (ADA), CoreWeave will ensure that qualified applicants and candidates with disabilities are provided reasonable accommodations for the hiring process, unless such accommodation would cause an undue hardship. If reasonable accommodation is needed, please contact: careers@coreweave.com.

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 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.

Eligible to access the export‑controlled information without a required export authorization, or

Eligible and reasonably likely to obtain the required export authorization from the applicable U.S. government agency. CoreWeave may, for legitimate business reasons, decline to pursue any export licensing process.

#J-18808-Ljbffr