Staff AI Engineer

2 months ago

USA (city not specified)Staff+

Base Salary

$205k - $307k/yr

Responsibilities

Define and evangelize the multi-year technical roadmap for MLOps.
Provide technical direction and mentorship across multiple engineering teams and organizations.
Design and architect scalable, extensible, performant, and reliable training, inference, and monitoring systems.
Partner with product, program, and data science leaders on cross-functional roadmaps.
Evaluate emerging MLOps technologies and establish engineering standards and best practices.
Apply sparsity, quantization, pruning, compression, and related optimization techniques to improve model performance and cost efficiency.
Collaborate with security teams on data protection and responsible AI practices.
Lead architecture reviews, publish engineering best practices, and mentor senior and lead engineers.

8+ years of professional software engineering experience.
5+ years of experience working in ML/AI or big data environments.
4+ years of technical leadership experience across multiple teams.
Expert-level proficiency in Python, Java, or Scala for production systems.
Hands-on experience with Docker, Kubernetes/Helm, and infrastructure-as-code tools such as Terraform or Pulumi.
Experience building and operating CI/CD pipelines for machine learning workflows using technologies such as Prefect or Airflow.
Experience deploying and operating secure, monitored services on AWS, Azure, or GCP.
Ability to mentor senior engineers, influence technical strategy, and drive cross-team initiatives to completion.
Preferred: Master’s or PhD in Computer Science, Engineering, Mathematics, or a related field.
Preferred: technical thought leadership demonstrated through open-source contributions, conference presentations, or publications.
Preferred: experience scaling ML platforms with distributed data technologies such as Spark or Kafka.
Preferred: expertise implementing compression, pruning, and quantization in production environments.

Hybrid/remote work arrangement.
Competitive benefits, DTO, parental leave, and an equity program.
The position includes a competitive base salary, annual performance bonus, and long-term incentives.
Opportunity to work on distributed, cloud-native systems and AI technology in high-stakes legal applications.
Opportunities for end-to-end ownership, knowledge sharing, continuous learning, and collaboration in an inclusive environment.