M

Senior Engineer, MAX Tools (Developer Tooling for AI Inference Stack)

Full Remote
Apply Now

Build developer tooling for Modular’s AI inference stack, focusing on profiling, debugging, and optimization tools for MAX framework. Design agent-native workflows and integrate tools across hardware platforms. Requires 4+ years of software engineering experience with systems/performance languages (C++, Rust, Go) and Python.

Key Highlights
Build developer tooling for AI inference stack (MAX framework)
Design agent-native workflows and tools for profiling, debugging, and optimization
Collaborate with kernel, model, and serving engineers to address pain points
Key Responsibilities
Enhance MAX developer experience with tools for traditional and agentic workflows
Design, build, and maintain profiling, tracing, debugging, and performance monitoring tools
Create agent-native tooling for model profiling, triage, and optimization
Integrate MAX tools across the stack with Mojo and driver APIs
Participate in design discussions and code reviews to uphold engineering standards
Collaborate with kernel, model, and serving engineers to address recurring pain points
Technical Skills Required
C++ Rust Python
Benefits & Perks
Annual target bonus
Equity
Benefits package
Equal opportunity and safe work environment
Remote work option (with in-person onboarding in Edinburgh)
Nice to Have
Experience with GPU performance tooling (Nsight Systems/Compute, rocprof, Perfetto)
Knowledge of AI frameworks like PyTorch, JAX, or TensorFlow
Experience building tooling for AI agents (e.g., agent skills, MCP)

Job Description

About Modular

At Modular, a Qualcomm company, we’re on a mission to revolutionize AI infrastructure by systematically rebuilding the AI software stack from the ground up. Our team, made up of industry leaders and experts, is building cutting-edge, modular infrastructure that simplifies AI development and deployment.

About The Role

At Modular we are building a next generation AI platform to power modern applications and facilitate access to cutting-edge hardware. The MAX framework is our developer-facing layer: it defines the APIs developers use to express models, integrate custom kernels, orchestrate execution, iterate on quality and ship systems into production.

Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

Remote Work Guides

Practical guides for finding and thriving in remote roles.

Browse all articles

Subscribe to our newsletter

New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.