M

Senior Software Engineer, MAX Tools

Relocation
This Job is No Longer Active This position is no longer accepting applications

Build and maintain developer tooling for Modular's AI inference stack, including profiling, debugging, and agent-native workflows. Design systems to support the MAX framework across heterogeneous hardware platforms and integrate with Mojo and driver APIs. Requires fluency in systems languages (C++, Rust, Go) and user-facing languages (Python), with experience in performance analysis and developer infrastructure.

Key Highlights
Develop agent-native tooling for AI model profiling, triage, and debugging.
Build profiling, tracing, and debugging tools for the MAX framework.
Integrate tooling across the AI stack, including Mojo and driver APIs.
Key Responsibilities
Enhance the MAX developer experience by creating tools for traditional and agentic workflows.
Design, build, and maintain profiling, tracing, debugging, and monitoring tools for the MAX framework.
Create agent-native tooling and workflows for AI agents to profile, triage, and debug issues.
Connect MAX tools through the entire stack by strengthening interfaces with Mojo and driver APIs.
Participate in design discussions and code reviews to uphold high engineering standards.
Collaborate with kernel, model, and serving engineers to address recurring pain points.
Technical Skills Required
C++ Rust Python
Benefits & Perks
Premier insurance plans
Up to 5% 401k matching
Flexible paid time off
Stock options
Relocation assistance for out-of-state US candidates
Nice to Have
Experience with GPU performance tooling (Nsight Systems/Compute, rocprof, Perfetto)
Knowledge of GPU programming (CUDA, ROCm)
Familiarity with AI frameworks like PyTorch, JAX, or TensorFlow
Experience building tooling for AI agents (e.g. agent skills, MCP)
Familiarity with Mojo

Job Description

At Modular, 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. By rethinking the complexities of AI systems, we’re empowering everyone to unlock AI’s full potential and tackle some of the world’s most pressing challenges.If you’re passionate about shaping the future of AI and creating tools that make a real difference in people’s lives, we want you on our team. You can read about our culture and careers to understand how we work and what we value.
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

Hidden company

Log in to see location

Premium

Systems Integration Engineer - Driver Safety Systems

Log in free to see the details

Hidden company

Log in to see location

Premium

Software Engineer

Log in free to see the details

Subscribe to our newsletter

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