Senior Machine Learning Engineer
Summary
Senior ML engineer builds and improves production-grade AI systems for a consumer-facing assistant, using Python, PyTorch/JAX, and GPU infrastructure.
About Us
More than 5 billion people use everyday applications such as email, notes, and task managers. Most of these tools are not built with AI. We are developing an AI assistant that helps users with conversations, daily tasks, planning, and work. It is easy to use and needs very little input from users.
Our product is designed to support long tasks, remember important information, and help users complete real work. It can solve problems step by step, connect with external tools, and provide reliable results even when AI responses are not always predictable. Our goal is to help people complete everyday tasks much faster while making the experience simple and enjoyable.
Role
As a Senior Member of Technical Staff – Machine Learning, you will take responsibility for important machine learning systems used in production. You will solve complex problems, design practical solutions, and build reliable systems that work well at scale.
This is a hands-on role for engineers who enjoy building and improving machine learning systems used by real customers.
Main Responsibilities
Build the main machine learning systems that support our AI product.
Manage the complete machine learning process, including data preparation, training, evaluation, inference, and continuous improvement.
Turn research ideas into production-ready machine learning solutions.
Find and fix model and system problems using data from production.
Improve systems through regular testing, measurement, and updates.
Work closely with research, engineering, and product teams to deliver useful AI features.
Support other machine learning engineers by sharing knowledge, reviewing work, and giving technical guidance.
Build systems that meet production requirements for speed, cost, reliability, and safety.
Technologies
Python
PyTorch / JAX
GPU-based training and inference systems
Requirements
Experience building and delivering machine learning systems used in production.
Good understanding of how modern machine learning models perform in real-world environments.
Ability to write clean, reliable, and maintainable production code.
Strong ownership and the ability to work independently from planning to delivery.
Fast learner with good communication skills and a mindset of continuous improvement.
Expected Results
Machine learning models meet goals for accuracy, speed, reliability, and efficiency in production.
Production problems are monitored, investigated, and solved quickly with minimal impact on users.
Data pipelines, training systems, and inference services remain reliable, scalable, and easy to maintain.
Machine learning systems improve over time based on real user feedback and production data.
Other engineers receive technical support and mentoring that helps improve the team's overall skills.
Work closely with different teams to make sure AI features fit well into the product and support business goals.
How We Work
We believe the best products are created by small teams with excellent skills. We work together, make decisions as a team, and move quickly while keeping high quality. We encourage everyone to stay organized, make good decisions, and work independently. Our goal is to build an AI product that provides real value to users every day.
Interview Process
If your experience matches our needs, we will invite you to 3 or 4 interviews.
Our technical team reviews every application. Interviews are held online or in person.
We keep the hiring process clear and efficient. If you show the skills and attitude we are looking for, we will invite you to join our team. We are looking for people who want to help build AI products that improve everyday life for billions of users.