AI Framework Engineer, Modular

AI Framework Engineer, Modular

Company Modular
Job title AI Framework Engineer, Graph Modeling
Job location United States / Canada
Type Full Time

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

Requirements & Skills:

  • 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

apply for job button