Staff Software Engineer, AI/ML
DigitalOcean
Staff Software Engineer, AI/ML Dive in and do the best work of your career at DigitalOcean. Journey alongside a strong community of top talent who are relentless in their drive to build the simplest scalable cloud. If you have a growth mindset, naturally like to think big and bold, and are energized by the fast-paced environment of a true industry disruptor, you'll find your place here. We value winning together—while learning, having fun, and making a profound difference for the dreamers and builders in the world. Building AI agents that take real actions is the easy part. Building agents that get better over time — that learn from feedback, correct mistakes, and optimize toward outcomes users actually care about — is one of the hardest open problems in production AI today. That's what this team works on. As a Staff AI/ML Engineer on our Applied Research team, you'll own the technical direction for feedback-driven learning in DigitalOcean's agentic systems: reward modeling, preference optimization, reinforcement learning, and the evaluation infrastructure needed to measure whether any of it is actually working. This is a senior IC role with broad technical scope. You'll set direction, run experiments at scale, and close the loop between user signals and model behavior - shipping research into production, not just writing it up. What You'll Be Doing Own the feedback learning roadmap Define and execute the applied research agenda for feedback-driven agentic AI — from reward modeling and preference optimization to online learning and human feedback loops. Translate user feedback, human evaluation data, and product signals into concrete training and optimization strategies. Stay close to the research frontier on RLHF, RLAIF, DPO, PPO, GRPO, and related methods and know when to apply them versus when simpler approaches win. Build production learning systems Design and implement learning loops that improve agent reasoning, planning, tool use, and action execution over time. Build evaluation frameworks that measure what matters: reasoning quality, instruction following, task success, safety, and real user outcomes — at both offline and online scale. Run large-scale experiments that connect model changes to measurable improvements in user experience and business impact. Provide technical leadership Set technical direction across modeling, experimentation strategy, evaluation design, and production readiness — without requiring direct management authority. Partner closely with product, engineering, design, and research teams to move work from prototype to shipped capability. Communicate complex AI systems clearly to both technical and non-technical stakeholders. What You'll Add to DigitalOcean We're looking for engineers who have shipped real learning systems — not just prototyped them. You likely bring: 8+ years of experience building production AI/ML systems — LLMs, GenAI, agentic systems, recommendation, search, personalization, or applied research at scale. Hands-on experience improving AI systems through reinforcement learning, reward modeling, fine-tuning, human feedback, or preference optimization — with results you can point to. Strong understanding of agentic AI: reasoning, planning, tool use, action execution, instruction following, and self-correction. Strong software engineering in Python and at least one production systems language. The judgment to balance model quality, product impact, latency, reliability, cost, and maintainability — and communicate those tradeoffs clearly. Preferred Qualifications Strong signal Experience with agent evaluation, offline/online experiments, and human feedback loops in production. Direct experience with RLHF, RLAIF, DPO, PPO, GRPO, or related optimization techniques. Prior Staff, Senior Staff, Tech Lead, or equivalent senior IC experience. Nice to have Master's or PhD in CS, ML, AI, or a related field — or equivalent depth demonstrated through industry work. Experience with production ML infrastructure: model serving, observability, data pipelines, feature stores, or experimentation platforms. Research contributions via publications, patents, open-source work, or demonstrated applied research impact in RL, reward modeling, evaluation, or recommendation systems. Compensation Range: $271,000 - $216,800 *This is a hybrid role
JR: 2026-7947
#LI-Hybrid Why You'll Like Working for DigitalOcean We innovate with purpose. You'll be a part of a cutting-edge technology company with an upward trajectory, who are proud to simplify cloud and AI so builders can spend more time creating software that changes the world. As a member of the team, you will be a Shark who thinks big, bold, and scrappy, like an owner with a bias for action and a powerful sense of responsibility for customers, products, employees, and decisions. We prioritize career development. At DO, you'll do the best work of your career. You will work with some of the smartest and most interesting people in the industry. We are a high-performance organization that will always challenge you to think big. Our organizational development team will provide you with resources to ensure you keep growing. We provide employees with reimbursement for relevant conferences, training, and education. All employees have access to LinkedIn Learning's 10,000+ courses to support their continued growth and development. We care about your well-being. Regardless of your location, we will provide you with a competitive array of benefits to support you from our Employee Assistance Program to Local Employee Meetups to flexible time off policy, to name a few. While the philosophy around our benefits is the same worldwide, specific benefits may vary based on local regulations and preferences. We reward our employees. The salary range for this position is based on market data, relevant years of experience, and skills. You may qualify for a bonus in addition to base salary; bonus amounts are determined based on company and individual performance. We also provide equity compensation to eligible employees, including equity grants upon hire and the option to participate in our Employee Stock Purchase Program. DigitalOcean is an equal-opportunity employer. We do not discriminate on the basis of race, religion, color, ancestry, national origin, caste, sex, sexual orientation, gender, gender identity or expression, age, disability, medical condition, pregnancy, genetic makeup, marital status, or military service. Application Limit: You may apply to a maximum of 3 positions within any 180-day period. This policy promotes better role-candidate matching and encourages thoughtful applications where your qualifications align most strongly. DigitalOcean$186.4k - $233k
...world. We are looking for a Staff Product Manager to own the... ...execution of DigitalOcean's Data and AI platform. This is a high-... ...technical execution — bridging Engineering, Infrastructure, and Go-to-Market... ...can spend more time creating software that changes the world. As a...SuggestedFull timeLocal areaWorldwideFlexible hours$209.1k - $282.9k
Software Engineer AI Inference Runtime Team As a Software Engineer on our AI Inference Runtime team, you will set technical direction for critical... ...+ years of experience, or equivalent demonstrated impact, in ML systems, high-performance systems, compilers, kernel...SuggestedWork at officeLocal area$189.3k - $290k
Staff Software Engineer Work Arrangement: This role is categorized as Remote/hybrid. This means the successful... ...for our next-generation Vehicle AI Assistant. In this role, you will lead the... ...Artificial Intelligence / Machine Learning (AI/ML) services. Establish and scale automated...SuggestedRemote workRelocation package$209.1k - $282.9k
We seek a Software Engineer to contribute to the next generation of Physical AI platforms on Arm. Your work will focus on building foundational software for AI tasks such... ...Linux, or real-time systems. Exposure to AI/ML inference systems, even from a platform or infrastructure...SuggestedWork at officeLocal area- Plaid In The Fintech Ai Ecosystem Plaid is uniquely positioned to become the financial... ...systems to life. Work with other AI engineers, software engineers and machine learning engineers... ...Haves Experience training and/or serving ML models in production, or fine-tuning LLMs...Suggested
$154k - $220k
...believe the future of work is Human + AI and are building an AI-native enterprise... .... Role We are looking for a Sr. Staff Software Development Engineer-AI Security to join our team. This is... ...Qualifications) Foundational understanding of AI/ML technologies and experience leveraging...Full timeWork at officeLocal area- Staff Software Engineer The power grid is changing — and the software behind it needs to change faster.... ...platforms and artificial intelligence (AI) agent frameworks that make next-generation... ...that orchestrate machine learning (ML) models, rule engines, simulation engines...
$110k - $260k
...Role Matters GEICO is re-engineering underwriting at scale. As a Senior Staff Engineer , you will play... ...scale mission-critical, AI-enabled systems while helping... ...integrating AI/ML solutions , including LLMs... ...0+ years of professional software engineering experience, including...Hourly payWork experience placementLocal area$209.1k - $282.9k
AI Compute Infra Engineer As an engineer on the AI Compute Infra team, you will design, build, and operate large-scale infrastructure for AI training... ..., with an interest in developing reliable infrastructure software. • Practical knowledge of Kubernetes, containers, Linux,...Work at officeLocal areaVisa sponsorshipRelocation package$182k - $242k
CoreWeave is The Essential Cloud for AI™. Built for pioneers by... ...: The Applications Engineering team in the Data Infrastructure... ...'s internal data-native software products, transforming data into... ...Experience developing and deploying AI/ML/LLM-powered applications in...Permanent employmentFull timeTemporary workCasual workWork at officeFlexible hours$170k - $230k
...About Us and the Role At Sobek AI, we’re building secure AI infrastructure for agentic... ...customer environments. We’re hiring a Software Engineer to help build the production AI systems... ...research role, it does require practical ML and LLM fundamentals. You should understand...Full timeVisa sponsorship$117.75k - $195k
..., and so are the rewards. We're looking for an exceptional Software Engineer, Agentic AI to help shape the future of our core platforms, products, and... ...to enhance compute and data efficiency across the ML infrastructure stack. Prototype and evaluate new Al tools,...Full timeWork at officeRemote workFlexible hoursShift work3 days per week$166k - $203k
...). The HackerOne Platform unites agentic AI solutions with the ingenuity of the world... ...inclusion, respect, and accountability. Software Engineer, Applied AI Location: Seattle, WA;... ...into applications Hands-on experience with ML frameworks such as PyTorch, TensorFlow,...Full timeApprenticeshipWork at officeLocal areaRemote workFlexible hoursShift work1 day per week$185k - $275k
...CoreWeave is The Essential Cloud for AI™. Built for pioneers by... .... What You’ll Do As a Staff Engineer, you will be a technical leader... ...You Are ~8+ years of software engineering experience. ~ Proven... ..., GPU-based applications, or ML pipelines. Knowledge of...Permanent employmentFull timeTemporary workCasual workWork at officeRemote workFlexible hours$131.6k - $210.3k
...Opportunity: We are seeking a Staff Software Engineer (Backend) to design, build, and operate... ...including services enhanced with Generative AI and agent-based capabilities. In this... ...Jasmine, Jest). • Prior experience with AI/ML-enabled systems in production...Full timeWork experience placementWork at officeLocal area$188.7k - $258.39k
...is committed to building breakthrough software with a spark of magic. We believe a... ...are looking for a Principal Software Engineer to join our Applied AI team. This is a rare opportunity to work... ...skills — you work effectively with ML researchers, data scientists, product...Full timeImmediate startFlexible hours- ...proprietary infrastructure and software, we empower over 250,000... ...stuff done” end-to-end. You use AI to work smarter and solve problems... ...coordinated fraud rings, ML models scoring every transaction... ...experienced team members in software engineering practices and coding standards...Full timeWorldwide
- ...usher in this new era, we seek AI-native thinkers across every... ...it didn’t stop there. They engineered Snowflake to power the Data Cloud... ...build data agents and create ML workflows using a... ...Principal engineers. AS A STAFF SOFTWARE ENGINEER - IDENTITY & ACCESS...Full time
$182k - $242k
...Description CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers,... ...more at About the Role As a Senior Software Engineer II (IC4) on the AI Workload... ...Familiarity with GPU-based workloads, ML training, or inference pipelines Knowledge...Permanent employmentFull timeTemporary workCasual workWork at officeFlexible hours$167.85k - $209.75k
...Sr. Software Engineer – AI BioDesign The Allen Institute accelerates science for a healthier world through large-scale research designed to answer... ...AI – inventing computational methods and large-scale AI/ML models that turn complex biological data into scientific insight...Work at officeLocal areaRemote workVisa sponsorshipWork visaRelocation package$61k - $101k
...require formal training or certification in software engineering concepts, along with 5+ years of applied... ...hands-on experience delivering agentic AI to production, beyond LLM wrappers or... ...Lead Software Engineer role within our AI/ML Data Platforms organization and...Full time$130k - $170k
...Founded by CPAs, tax attorneys, and engineers, Taxbit is the leading... ...the digital economy. Taxbit's AI-enabled platform streamlines compliance... ...quality and distribute AI/ML knowledge across the team... ...~3+ years of professional software engineering experience, with at...Full timeWork at officeWork from homeFlexible hours$168.75k
...with our ecosystem of devices and cloud software. Like our products, we work better... ...where you matter. Your Impact As a Staff Embedded Software Engineer, you will lead critical software engineering... ...connected devices, influencing how AI models are trained, deployed,...Work experience placementWork at officeRemote work$207k - $275k
CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave... ...About the role As a Staff Software Engineer, you will define and drive the technical... ...benchmarking, Kubernetes infrastructure, and AI/ML platform reliability. This role...Permanent employmentFull timeTemporary workCasual workWork at officeFlexible hours$163.9k - $235.55k
...exceptionally skilled and visionary Principal Software Engineer with deep expertise in building, deploying, and scaling generative AI applications. As a key individual... ...mission. The Principal Software Engineer, AI/ML will lead the technical vision, strategy, and...Full timeLocal area$209.1k - $282.9k
As a software Engineer on the AI Compute Platform team, you will design and build a secure, reliable, and easy-to-use platform for running AI workloads... ...or production backend servicesExperience with distributed ML training or inference systemsStrong programming skills in...Work at officeLocal areaVisa sponsorshipRelocation package$87.36k - $124.8k
...the agentic enterprise. To usher in this new era, we seek AI-native thinkers across every function who are energized by... ..., but to help redefine the future of how work gets done. Software Engineer Intern (AI / ML) - Spring 2027 Snowflake started with a clear vision:...Full timeCasual workInternshipWork from homeFlexible hours- ...Python Software Engineer - AI Workflows About the Role What if your Python expertise could directly shape the infrastructure behind next... ..., data quality, or evaluation systems Familiarity with AI/ML workflows, model training, or benchmarking pipelines...Hourly payOngoing contractContract workFreelanceRemote workFlexible hours
- ...usher in this new era, we seek AI-native thinkers across every... ...-paced, collaborative team of engineers on a journey to provide customers... ...at Snowflake. AS A SENIOR SOFTWARE ENGINEER IN DATA CLEAN ROOMS,... ...capabilities and support custom AI/ML workflows. Partner closely...Full timeShift work
- ...significant impact in executing this mission. The ML Platform team at Zoox plays a crucial... ...will work alongside a team of strong software engineers and act as a force multiplier for our... ...We may use artificial intelligence (AI) tools to support parts of the hiring process...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Software Engineer, AI/ML. Be the first to apply!
- software technology Seattle, WA
- remote software sales Seattle, WA
- embedded software Seattle, WA
- software intern Seattle, WA
- software support Seattle, WA
- ultimate software Seattle, WA
- software engineer - cloud services Seattle, WA
- entry level software sales Seattle, WA
- software trainer Seattle, WA
- software Seattle, WA



