Senior Software Engineer, W&B Agent - W&B
About The Position
As a Senior Full Stack Engineer on the ARIA Agent team at Weights & Biases at CoreWeave, you will build and maintain ARIA, an AI research and iteration agent designed to enable continuous experimentation at scale. The team is building the research agent from scratch, spanning distributed backend systems that execute multi-turn agent conversations and the React user interface used by ML practitioners. The goal is to deliver a fast, reliable, and intuitive agent experience that accelerates machine-learning workflows. You will own features and systems across the stack, including backend infrastructure and API design, distributed worker systems, streaming architecture for real-time token output, and frontend components for the chat interface, billing, and agent configuration. You will collaborate with ML science, product, iOS, and SRE teams, combining reliability engineering with product sensibility. Work may include debugging production memory issues in Kubernetes workers, delivering UI components, and strengthening security controls for code-execution sandboxes. Engineers are accountable for outcomes end to end. The role is part of CoreWeave, which acquired Weights & Biases to combine cloud infrastructure with AI development tools across the AI lifecycle. CoreWeave describes its culture as collaborative, entrepreneurial, fast-moving, and focused on innovation, learning, and delivering strong client experiences. The posted base salary range is $182,000–$242,000, with starting compensation determined by job-related knowledge, skills, experience, and market location. The total rewards package may also include a discretionary bonus, equity awards, and benefits, subject to eligibility.
Requirements
- Bachelor’s degree or foreign equivalent in Computer Science, Computer Engineering, Data Science, Information Systems, or a closely related technical or quantitative field involving substantial engineering coursework.
- At least 4 years of full-stack engineering experience, with meaningful depth in backend services and React/TypeScript frontend development.
- Strong Python backend skills, including async patterns, FastAPI, distributed state machines, and Kubernetes pod lifecycle.
- Proficiency in TypeScript and React, including production UI components, real-time streaming interfaces, complex state management, and responsive web and mobile-web layouts.
- Experience building and operating services at scale, including rolling deployments, autoscaling, worker concurrency, and graceful termination.
- Experience with observability tooling such as Datadog, Weave tracing, or structured logging, and a track record of building dashboards that surface actionable signals.
- Security-conscious engineering experience or mindset, including input validation, sandboxed code execution, egress controls, and payload hardening.
- Experience with MySQL or a similar relational database management system for transactional state management.
- Must meet applicable U.S. export-control eligibility requirements for access to export-controlled information.
Nice To Haves
- Experience building streaming user interfaces using server-sent events or WebSockets, including graceful fallback handling.
- Background in ML infrastructure, developer tooling, or observability platforms.
- Hands-on experience building and shipping production-grade AI agents, including tool use, multi-turn conversation management, and agent reliability patterns.
- Comfort working in a fast-paced startup environment, moving quickly, making pragmatic tradeoffs, and iterating on user feedback while maintaining production quality.
Responsibilities
- Build the ARIA research agent and systems that support continuous experimentation at scale.
- Own features and systems across backend infrastructure, API design, and React frontend development.
- Build and maintain distributed worker systems that run agent conversations at scale.
- Design streaming architecture that delivers real-time token output to the browser.
- Contribute frontend components for the chat interface, billing UI, and agent configuration experience.
- Operate and improve production services, including debugging Kubernetes worker issues and improving deployment, autoscaling, concurrency, and graceful termination behaviors.
- Strengthen security controls for code execution, including sandbox egress controls and input or payload hardening.
- Collaborate with ML science, product, iOS, and SRE teams.
- Build observability dashboards and tools that surface actionable signals, and use measurement and iteration to improve system performance and reliability.
Benefits
- Discretionary bonus and equity awards, subject to eligibility.
- Medical, dental, and vision insurance, 100% paid for by CoreWeave for the described US-based full-time employee offering.
- 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 the Employee Stock Purchase Program (ESPP).
- Mental wellness benefits through Spring Health.
- Family-forming support through Carrot.
- Paid parental leave.
- Flexible, full-service childcare support with Kinside.
- 401(k) with employer match.
- Flexible PTO.
- Catered lunch each day at office and data center locations.
- Casual work environment.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
