Staff Software Engineer, SDK

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

As a Staff Software Engineer on the SDK team at Weights & Biases, you'll own core parts of the open-source W&B SDK and the systems that carry experiment data from a user's training script into the W&B platform. This is foundational product surface area used every day by everyone from individual researchers to some of the largest AI teams in the world.

Your Work May Span

• The SDK, CLI, and surrounding developer workflows that make W&B intuitive, dependable, and fast in real-world ML development.
• Performance-sensitive runtime systems that capture, buffer, and move data reliably across local, cloud, and large-scale training environments.
• The experiment data path end-to-end, with an emphasis on correctness, resilience, and operability.
• Tooling and observability that help us understand how the product is used, diagnose issues quickly, and improve the user experience with confidence.
  
  

At the Staff level, you'll help set technical direction, partner closely with backend, platform, and product teams, talk directly with users, and mentor other engineers.

Who You Are

• 7 years of professional software engineering experience (Staff candidates: typically 10 ), with meaningful time spent building and operating production systems in Python and/or Go.
• A strong track record of designing and evolving non-trivial systems with real performance, reliability, and user impact.
• Comfort working across boundaries between developer tools, runtimes, networked services, and data systems.
• Experience building reliable systems or pipelines, with good instincts around correctness, resilience, observability, and operational simplicity.
• An end-to-end ownership mindset: you can scope problems, make sound tradeoffs, ship, measure, and iterate.
• Comfort working in an open, user-facing environment where feedback is direct and quality matters.
  
  

Preferred

• Hands-on experience with ML workflows, model training, or the tooling researchers use day to day.
• Experience with distributed systems, performance-sensitive infrastructure, or large-scale training environments.
• Experience with systems instrumentation, observability, or infrastructure that has to work well outside the happy path.
• Experience building CLIs, local-first developer tools, or terminal-based workflows.
• Prior work on open-source libraries, SDKs, or developer platforms with broad external adoption.
  
  

Wondering if you’re a good fit? We believe in investing in our people, and value candidates who can bring their own diversified experiences to our teams – even if you aren't a 100% skill or experience match. Here are a few qualities we’ve found compatible with our team. If some of this describes you, we’d love to talk.

• Care deeply about users: You want to talk to real ML researchers and platform engineers, hear about their pain, and go fix it.
• Autonomous: You work well in a self-directed environment, proactively find ways to improve things, and can drive a project across team boundaries without needing to be told.
• Curious and driven: You're interested in how ML practitioners actually work — the frameworks, the hardware, the failure modes — and you want to keep learning.
• Pragmatic: You know when to build the right thing and when to ship the thing that unblocks a customer this week. You can hold both in your head at once.
  
  

Why Us?

Top-tier AI/ML teams rely on the SDK this team ships, every day. The code you write here runs on the hardware training many of the models you've heard of. The W&B SDK is one of the most widely adopted open-source libraries in the ML ecosystem, and it sits at a unique vantage point on the industry: we see, in aggregate, how the frontier is actually training models. You'll get to shape that, in public, alongside a small team that cares about craft.

About

We work hard, have fun, and move fast. 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
  
  

We support and encourage an entrepreneurial outlook and independent thinking. We foster an environment that encourages collaboration and provides the opportunity to develop innovative solutions to complex problems. As we get set for takeoff, the growth opportunities within the organization are constantly expanding. You will be surrounded by some of the best talent in the industry, who will want to learn from you, too. Come join us!

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. We strive for both market alignment and internal equity when determining compensation. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).

What We Offer

The range we’ve posted represents the typical compensation range for this role. To determine actual compensation, we review the market rate for each candidate which can include a variety of factors. These include qualifications, experience, interview performance, and location.

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

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

California Consumer Privacy Act - California applicants only

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) 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, (B) eligible to access the export controlled information without a required export authorization, or (C) 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.