Lead applied ML research, experimentation, and production model development to optimize auctions and decisioning in Rokt's multi-sided ecommerce marketplace, powering billions of transactions. Requires PhD (or equiv), 10+ years building production ML systems, and deep expertise in at least two advanced ML areas such as deep learning architectures, RL, or revenue optimization.
470k – 615k/yr
Hybrid10+ YOEML Engineering
About the role
Responsibilities
Technical
Own the research, design, development, testing, deployment and maintenance of machine learning systems and services at Rokt.
Optimise auction and decisioning logic, applying model predictions to maximise value across Rokt’s multi-sided marketplace.
Conduct applied ML research, prototype new modeling approaches, and rigorously test innovations.
Evaluate and improve model performance, ensuring robustness, scalability, and interpretability.
Translate complex business needs into practical ML solutions, collaborating with product and engineering teams.
Stay ahead of emerging ML trends, contributing to knowledge sharing through tech talks, brown bags, and best practice evangelism.
Leadership
Act as technical and project leader of a small team of specialists.
Provide leadership on complex technical issues.
Set standards for technical excellence - from coding to architectural best practices.
Work cross-functionally with engineering and product leadership to plan and manage delivery expectations of roadmap items.
Actively monitor progress and intervene where required to mitigate risks and bottlenecks, managing expectations within and outside the team.
Requirements
PhD or equivalent experience in Computer Science, Statistics, Mathematics, or related field with specialization in ML, AI, or Information Retrieval, and experience leading PhD level scientists/engineers on projects.
10+ years of industry experience building production-grade ML systems.
Technical leader building Claude Science, an AI workbench for researchers that integrates literature, databases, scientific computing, analysis, and publication tools. Shape product and architecture from 0→1, partner with scientists and research teams to translate domain needs into shipped features and improved model capabilities.
405k – 485k/yr
Hybrid8+ YOEML Engineering
Staff+ Software Engineer, Inference Runtime
AnthropicSan Francisco, CA +2
Technical lead for the shared, accelerator-agnostic inference runtime serving Claude. Owns architecture, performance, and validation for GPU/TPU/Trainium platforms in a high-scale distributed systems environment.
405k – 485k/yr
Hybrid8+ YOEML Engineering
Staff Research Engineer
DecagonSan Francisco, CA +1
Staff Research Engineer leads development of conversational AI models for enterprise support agents, owning multi-quarter initiatives from research to production deployment. Requires 8+ years in AI/ML with experience deploying LLMs, Python fluency, and cross-functional leadership.
325k – 425k/yr
On-site8+ YOEML Engineering
Member of Technical Staff
xAIPalo Alto, CA
Member of Technical Staff at xAI performing cutting-edge AI research and hands-on engineering to advance large language models, quantitative reasoning, and language understanding. Requires Bachelor's in CS/ML-related field plus 2 years experience with big data tools, Kubernetes, deep learning, NLP/CV/speech, and languages like Rust/C++/Python.
324k – 396k/yr
On-site2+ YOEML Engineering
Member of Technical Staff
xAIPalo Alto, CA
Hands-on technical leader building and scaling large language models and AI systems. Requires 3-5+ years of AI/ML experience with strong Python and deep learning frameworks.