Staff Software Engineer - AI Platform

The role

You will architect and lead the development of scalable agentic systems using LLMs to automate workflows. You will define the technical vision for LLM integration and set engineering standards for distributed machine learning solutions.

What you will do

• Architect and lead the development of highly scalable intelligent agents that utilize LLMs
• Own and drive the technical vision and long-term strategy for integrating LLMs
• Partner with leadership to set the strategic direction for cognitive automation
• Take ownership of the development lifecycle for machine learning-powered tools
• Implement and enforce industry-leading standards for building production-grade distributed machine learning solutions
• Research and experiment with emerging trends in machine learning and AI agents
• Work with Machine Learning engineers and product teams to deliver impactful solutions
• Provide guidance to engineering levels from mid-level to senior leads
• Contribute to hiring strategy and technical bar-raising

Required qualifications

• Bachelor's, Master's degree or Ph.D. in Computer Science, Machine Learning, or a related field
• 12+ years of hands-on experience in software engineering
• Experience designing and building large-scale distributed systems
• Experience building and deploying complex agentic applications leveraging LLMs in production environments
• Track record of org-wide technical impact
• Proficiency in Python or similar programming languages
• Deep expertise with cloud and container frameworks such as AWS and Kubernetes
• Track record of designing and building highly scalable, fault-tolerant, and observable distributed systems
• Expert-level grasp of machine learning principles, algorithms, and evaluation metrics
• Software engineering best practices for distributed systems
• Success in owning the design, deployment, and operation of scalable backend software in high-stakes production environments

Nice to have

• Expertise in Golang is strongly preferred

Benefits

• Competitive base salary
• Annual company performance bonus program
• Equity in the form of Restricted Stock Units (RSUs)
• Health coverage
• Retirement savings
• Time off
• Family planning programs

About the company

Rivian and Volkswagen Group Technologies is a joint venture between two industry leaders. They address the challenges of electric vehicles through technology for software-defined vehicles.