AI Technical Writer, Modular

$166.5-242k

+ Bonus and equity. For candidates based in Canada, the base salary range is $157,500 - $228,500 CAD, plus bonus and equity

Python

Tensorflow

PyTorch

Sagemaker

Mid and Senior level

Remote from Canada, US

Fast, scalable Gen AI inference platform

Job no longer available

Fast, scalable Gen AI inference platform

201-500 employees

B2BArtificial IntelligenceMachine Learning

Job no longer available

$166.5-242k

+ Bonus and equity. For candidates based in Canada, the base salary range is $157,500 - $228,500 CAD, plus bonus and equity

Python

Tensorflow

PyTorch

Sagemaker

Mid and Senior level

Remote from Canada, US

201-500 employees

B2BArtificial IntelligenceMachine Learning

Company mission

To have real, positive impact in the world by reinventing the way AI technology is developed and deployed into production with a next-generation developer platform.

Job

Company

Role

Who you are

2+ years experience writing code that uses machine learning products such as PyTorch, TensorFlow, ONNX, Vertex AI, SageMaker, etc
3+ years experience writing tutorials, blogs, API references, or other technical documents for software engineers
Strong written and verbal communication skills, including the ability to explain complex technical concepts in simple language
Strong research and analytical skills, including the ability to gather and organize information from a variety of sources
A strong curiosity to learn through personal exploration and from our customers, team, and industry
On the spectrum between a pure writer and engineer, we want somebody with strong engineering skills
You must be comfortable writing code and troubleshooting new technologies, and also be excited to teach other people about those technologies

What the job involves

We’re looking for somebody with significant experience writing docs and example code for ML developers
Experiment with new API libraries to build and deploy AI models, writing code examples yourself and/or collaborating with engineers to design examples for critical use cases
Collaborate with the product and engineering teams to define the user experience and identify issues that should be fixed or documented
Write developer guides, tutorials, concept guides, and other documentation for topics that use MAX APIs in multiple programming languages, including Python, C, and Mojo
Maintain existing code examples and documentation, including editing and revising documentation to ensure completeness, correctness, and clarity
Collaborate with other technical writers on large documentation sets, using a shared structure and style guide

Our take

AI development is booming, but the tools behind it? Not so much. Developers are often stuck dealing with messy, disconnected systems that slow everything down and bump up costs.

To address this, Modular has developed a unified AI platform that streamlines the development process. It's AI Engine enhances the performance of models on CPUs and GPUs, supporting popular frameworks like TensorFlow and PyTorch. And on top of that, it's built Mojo, a programming language that's as easy as Python but much faster (35,000 times faster).

In 2023, the company raised an impressive $100M to power its mission. It's been using this funding to grow its team, support more hardware, and push Mojo even further. With the hopes of making AI development faster, smoother, and a whole lot more fun for everyone.

Kirsty

Company Specialist at Welcome to the Jungle

Company

Funding (last 2 of 3 rounds)

Sep 2025

$250m

LATE VC

Aug 2023

$100m

LATE VC

Total funding: $380m

Company benefits

A variety of fantastic health benefits (health, dental, vision insurance; life insurance etc) are available
A 401k plan with up to 5% match
Free tax advice on Carta
Generous work-from-home stipend of $1500 to help you improve your home office
Unlimited paid time off and flexible work hours

Leadership

Chris Lattner

(Co-Founder & CEO)

Founder, Architect, Engineer and BDFL for the LLVM Organization. Former President of Engineering and Product at SiFive, as well as working in various roles at Google, Tesla, and Apple.

Tim Davis

(Co-Founder & President)

Worked at Google for 5 years, ultimately as Group Product Lead for Google ML. Previously founded Fluc Inc, where they developed the entire ML powered delivery system with their co-founder, and mapping software system CrowdSend.

Share this job

View 14 more jobs at Modular