Machine Learning Engineer, On-Device ML

Posted 2 Days Ago
Easy Apply
San Francisco, CA
Senior level
Artificial Intelligence • Information Technology • Machine Learning • Natural Language Processing • Productivity • Software • Generative AI
Grammarly improves lives by improving communication.
The Role
As a Machine Learning Engineer on the On-Device ML team at Grammarly, you will design and implement ML models for on-device features, evaluate their performance, and collaborate with product teams to integrate these capabilities into user interfaces. You will leverage academic research to enhance the technology and launch models while responding to user feedback.
Summary Generated by Built In

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 30 million people and 70,000 teams. 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).

#LI-Hybrid

 

Top Skills

Rust

What the Team is Saying

Nicole
Maryna
Brad
Shanik
The Company
San Francisco, CA
1,000 Employees
Hybrid Workplace
Year Founded: 2009

What We Do

Grammarly’s digital writing assistant helps 30 million people and 70,000 teams write more clearly and effectively every day. In building a product that scales across multiple platforms and devices, Grammarly works to empower users whenever and wherever they communicate.

Why Work With Us

At Grammarly, we’re defining what a digital writing assistant can be. So far, we help 30 million people and 30,000 teams communicate every day, and we’re always pushing the possibilities of our product. With a globally connected team that collaborates with autonomy and trust, we’re focused on long-term success—and excited about our future.

Gallery

Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery

Grammarly Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Typical time on-site: Flexible
San Francisco, CA

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account