AI Engineer Researcher

AI Engineer Researcher – Weights & Biases

Join to apply for the AI Engineer Researcher role at Weights & Biases
, part of Core Weave.

Core Weave, the AI hyperscaler, has acquired Weights & Biases to deliver a powerful end‑to‑end platform that helps developers build, train, and deploy AI models faster. We are committed to accelerating the AI lifecycle—from experiment tracking and model optimization to high‑performance training clusters, agent building, and inference at scale.

Your responsibilities will include helping productize the continuous training loops our team develops and shaping the user experience for our customers. You will work in an environment of ambiguity, experiment rapidly, and drive impact from prototype to production.

• Productize continuous training loops and design user‑facing tools (dashboards, APIs, scripts).
• Collaborate closely with product and data engineering teams to iterate on feature design.
• Deliver high‑quality, well‑tested code that runs at scale (CUDA kernels, LLM tracing dashboards, etc.).
• Experiment and ship quickly, learning from failures and dropping what does not work.

We’re looking for engineers who learn fast, ship value, and own end‑to‑end solutions.

• Experience building large‑scale machine learning pipelines or agent systems.
• Proficiency in Python, CUDA, and cloud orchestration (Kubernetes). Familiarity with ML frameworks such as Megatron or Unsloth is a plus.
• Strong debugging, profiling, and performance‑engineering skills.
• Excellent communication and ability to work across distributed teams.
• Demonstrated impact in prior projects or open source contributions; formal education is considered less important than outcome.

• Python, CUDA
• Kubernetes, Temporal, Postgres
• Megatron, Unsloth, FastAPI

• Be Curious at Your Core
• Act Like an Owner
• Empower Employees
• Deliver Best‑in‑Class Client Experiences
• Achieve More Together

Base salary range: $188,000 – $275,000. Total rewards include discretionary bonus, equity awards, and a comprehensive benefits program.

• Medical, dental, and vision insurance (100% paid)
• Life, supplemental life, and disability insurance
• Flexible Spending Account & Health Savings Account
• Tuition reimbursement
• Employee Stock Purchase Program
• Mental wellness benefits through Spring Health
• Paid parental leave and childcare support
• 401(k) with generous employer match
• Flexible PTO and a casual work environment
• Catered lunch daily at our office and data center locations

Hybrid with remote options for candidates located more than 30 miles from an office. New hires will attend onboarding at one of our hubs within their first month.

Core Weave is an equal‑opportunity employer. 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.

This position requires access to export‑controlled information. Applicants must be U.S. persons or meet the criteria listed in the job posting.