Weights & Biases
AI Engineer- ML/RL – Weights & Biases
Weights & Biases, San Francisco, California, United States, 94199
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 The AI team is a hands‑on applied AI group at Weights & Biases that turns frontier research into teachable workflows. We collaborate with leading enterprises and the OSS community. We are the team that took W&B from a few hundred users to millions of users and one of the most beloved tools in the ML community.
This is a senior applied role at the research‑to‑production boundary. You will prototype, evaluate, and ship reusable DL/RL workflows for enterprise use on the W&B stack—then document and teach them to our customers and the community. The focus is application, not novel research: rapid prototyping, careful evaluation, and production‑grade reference implementations with clear trade‑offs.
About The Role
Understand the state‑of‑the‑art in deep learning / AI and turn the research into practical workflows that can be adopted by our users, the open source community & enterprise customers alike.
Build in public: Publish engineering artifacts (code, reports, talks) that teach how to reproduce results; engage with OSS and customer engineers.
Design and ship reference workflows for post‑training & agents (SFT/DPO/GRPO/PPO, reward models, online RLHF/RLAIF) with reproducible repos, W&B Reports, and dashboards others can run.
Own end‑to‑end demos: data >> distributed training (FSDP/ZeRO/DeepSpeed/JAX pjit) >> evaluation (lm‑eval‑harness + agent benches) >> serving (vLLM/TensorRT-LLM/Triton/SGLang).
Partner with lighthouse customers; turn recurring patterns into templates and product feedback.
Track recent advances (papers, releases, kernels), run focused ablations, and translate wins into production‑ready workflows.
Run growth experiments to track the usage of the Weights & Biases suite of products from the artifacts built.
Who You Are
Deep learning: 5+ years training large models in PyTorch or JAX; strong numerics (autograd, initialization, mixed precision).
RL/RLHF: hands‑on with SFT/DPO/GRPO/PPO, reward modeling, preference data pipelines, and online/offline RL for LLMs/agents.
Inference/serving: production experience with vLLM/TensorRT-LLM/Triton; quantization, speculative decoding, caching.
Evaluation: built task/agent harnesses with statistically sound metrics (variance, CIs, power) and failure taxonomies.
Systems: strong Python plus one: CUDA/Triton kernels, custom C++ ops, or high‑performance data ingestion.
Reproducibility: rigorous experiment tracking (sweeps, artifacts, lineage); minimal repros others can run.
Public signal: 2+ OSS repos/notebooks/talks with adoption (e.g., stars, forks, downloads, conference views).
Preferred
Paper‑to‑production within weeks at a top lab or applied‑AI team (pretrain >> post‑train >> eval >> serve).
Data engines & feedback loops (rater pipelines, synthetic data, active learning).
Prior customer enablement with external adoption at scale.
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.
You love turning cutting‑edge AI research into clean, benchmarked, production‑ready templates others can use today.
You’re curious about RL-based post‑training and agent evaluation, and maintain your own leaderboards.
You’re an expert in at least one of: distributed training at scale, RLHF/GRPO systems, or low‑latency LLM serving—and you can demonstrate it with code and benchmarks.
The base pay and target total cash for this position range from $182,000 to $242,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.
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.
#J-18808-Ljbffr
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 The AI team is a hands‑on applied AI group at Weights & Biases that turns frontier research into teachable workflows. We collaborate with leading enterprises and the OSS community. We are the team that took W&B from a few hundred users to millions of users and one of the most beloved tools in the ML community.
This is a senior applied role at the research‑to‑production boundary. You will prototype, evaluate, and ship reusable DL/RL workflows for enterprise use on the W&B stack—then document and teach them to our customers and the community. The focus is application, not novel research: rapid prototyping, careful evaluation, and production‑grade reference implementations with clear trade‑offs.
About The Role
Understand the state‑of‑the‑art in deep learning / AI and turn the research into practical workflows that can be adopted by our users, the open source community & enterprise customers alike.
Build in public: Publish engineering artifacts (code, reports, talks) that teach how to reproduce results; engage with OSS and customer engineers.
Design and ship reference workflows for post‑training & agents (SFT/DPO/GRPO/PPO, reward models, online RLHF/RLAIF) with reproducible repos, W&B Reports, and dashboards others can run.
Own end‑to‑end demos: data >> distributed training (FSDP/ZeRO/DeepSpeed/JAX pjit) >> evaluation (lm‑eval‑harness + agent benches) >> serving (vLLM/TensorRT-LLM/Triton/SGLang).
Partner with lighthouse customers; turn recurring patterns into templates and product feedback.
Track recent advances (papers, releases, kernels), run focused ablations, and translate wins into production‑ready workflows.
Run growth experiments to track the usage of the Weights & Biases suite of products from the artifacts built.
Who You Are
Deep learning: 5+ years training large models in PyTorch or JAX; strong numerics (autograd, initialization, mixed precision).
RL/RLHF: hands‑on with SFT/DPO/GRPO/PPO, reward modeling, preference data pipelines, and online/offline RL for LLMs/agents.
Inference/serving: production experience with vLLM/TensorRT-LLM/Triton; quantization, speculative decoding, caching.
Evaluation: built task/agent harnesses with statistically sound metrics (variance, CIs, power) and failure taxonomies.
Systems: strong Python plus one: CUDA/Triton kernels, custom C++ ops, or high‑performance data ingestion.
Reproducibility: rigorous experiment tracking (sweeps, artifacts, lineage); minimal repros others can run.
Public signal: 2+ OSS repos/notebooks/talks with adoption (e.g., stars, forks, downloads, conference views).
Preferred
Paper‑to‑production within weeks at a top lab or applied‑AI team (pretrain >> post‑train >> eval >> serve).
Data engines & feedback loops (rater pipelines, synthetic data, active learning).
Prior customer enablement with external adoption at scale.
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.
You love turning cutting‑edge AI research into clean, benchmarked, production‑ready templates others can use today.
You’re curious about RL-based post‑training and agent evaluation, and maintain your own leaderboards.
You’re an expert in at least one of: distributed training at scale, RLHF/GRPO systems, or low‑latency LLM serving—and you can demonstrate it with code and benchmarks.
The base pay and target total cash for this position range from $182,000 to $242,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.
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.
#J-18808-Ljbffr