Machine Learning Engineer, On-Device ML

Grammarly offers a dynamic hybrid working model for this role. This flexible approach gives team members the best of both worlds: plenty of focus time along with in-person collaboration that helps foster trust, innovation, and a strong team culture.            

About Grammarly

Grammarly is the world’s leading AI writing assistance company trusted by over 40 million people and 50,000 organizations. From instantly creating a first draft to perfecting every message, Grammarly helps people at 96% of the Fortune 500 and teams at companies like Atlassian, Databricks, and Zoom get their point across—and get results—with best-in-class security practices that keep data private and protected. Founded in 2009, Grammarly is No. 14 on the Forbes Cloud 100, one of TIME’s 100 Most Influential Companies, one of Fast Company’s Most Innovative Companies in AI, and one of Inc.’s Best Workplaces.

The Opportunity

To achieve our ambitious goals, we’re looking for a Senior ML Engineer to join our On-Device team. This role will join a talented and senior team of engineers and researchers working to bring the power of Grammarly’s LLM-powered writing assistance onto your computer. On-device is an integral part of Grammarly’s future, and we specifically want to find people with the talent and vision to help us unlock new ideas and product directions.

Grammarly’s engineers and researchers have the freedom to innovate and uncover breakthroughs—and, in turn, influence our product roadmap. The complexity of our technical challenges is growing rapidly as we scale our interfaces, algorithms, and infrastructure. You can hear more from our team on our technical blog.

As a Machine Learning Engineer on the On-Device ML team, you will:

• Propose, design, and prototype strategic new features that on-device capabilities make possible.
• Build ML models to solve real-world problems and leverage data to evaluate and improve their quality.
• Own the implementation of your models in an on-device context, taking into account considerations like model size, performance, and latency.
• Collaborate with Grammarly’s core product teams to integrate your models into our user interfaces.
• Launch and monitor the deployment of ML models via A/B testing and respond to issues that arise from user feedback and telemetry.
• Leverage academic research to ensure that our technology continues to reflect and advance the state of the art.

Qualifications

• Has a deep understanding of modern algorithms for and approaches to ML and NLP.
• Has expert-level productivity coding in a modern compiled language and is excited about developing in Rust.
• Has worked on multiple platforms (macOS, Windows, Linux, and Android), especially cross-platform projects.
• Has prior experience dealing with the constraints imposed by writing device-side code. (optional, but helpful)
• Has working knowledge of front-end development practices. (optional, but helpful)
• Is highly entrepreneurial, energetic, and creative, and is comfortable working in a fast-paced environment with high levels of ambiguity and evolving product requirements.
• Can communicate with other humans more effectively than a generative AI system can.
• Embodies our EAGER values—is ethical, adaptable, gritty, empathetic, and remarkable.
• Is inspired by our MOVE principles: move fast and learn faster; obsess about creating customer value; value impact over activity; and embrace healthy disagreement rooted in trust.

Compensation and Benefits

Grammarly offers all team members competitive pay along with a benefits package encompassing the following and more: 

• Excellent health care (including a wide range of medical, dental, vision, mental health, and fertility benefits)
• Disability and life insurance options
• 401(k) and RRSP matching 
• Paid parental leave
• 20 days of paid time off per year, 12 days of paid holidays per year, two floating holidays per year, and flexible sick time
• Generous stipends (including those for caregiving, pet care, wellness, your home office, and more)
• Annual professional development budget and opportunities

Zone 1: $337,000 - $403,000/year (USD)

We encourage you to apply

At Grammarly, we value our differences, and we encourage all—especially those whose identities are traditionally underrepresented in tech organizations—to apply. We do not discriminate on the basis of race, religion, color, gender expression or identity, sexual orientation, ancestry, national origin, citizenship, age, marital status, veteran status, disability status, political belief, or any other characteristic protected by law. Grammarly is an equal opportunity employer and a participant in the US federal E-Verify program (US). We also abide by the Employment Equity Act (Canada).