Modular is hiring a
AI Framework Engineer

Summary

Join our world-leading AI infrastructure team and help drive our industry forward! We are looking for candidates based on their breadth and depth of experience in software engineering, machine learning, compilers, hardware, and more. In the Max Engine SDKs team, we are focused on building software that helps drive Modular product adoption by delivering differentiated solutions to the industry’s most challenging AI deployment problems.

Requirements

• Experience building ML models in PyTorch and/or ONNX
• Familiarity with HuggingFace API and workflows for using community models
• Ability to create durable, reusable software tools and libraries that are leveraged across teams and functions
• Experience in machine learning technologies and use cases
• Creativity and curiosity for solving complex problems, a team-oriented attitude that enables you to work well with others, and alignment with our culture
• Strongly identifies with our core company cultural values

Responsibilities

• Work with Product and partner engineering teams to design and ship new SDK features
• Build serving-optimized LLMs and broader modeling libraries on top of our Max Graph API
• Contribute to and maintain our user-facing MAX benchmarking, development, and visualization tools and inference APIs

Preferred Qualifications

• In-depth knowledge of modern C++
• Some experience with and interest in Mojo, our AI programming language!

Benefits

• Premier insurance plans
• Up to 5% 401k matching
• Very generous WFH setup stipends
• Flexible paid time off