Full-Time

AI Framework Engineer

Graph Modeling

Posted on 4/30/2024

Modular

Modular

51-200 employees

Unified AI frameworks and hardware platform

AI & Machine Learning

Mid

Remote in USA + 1 more

Required Skills
Pytorch
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
  • Experience in machine learning technologies and use cases
  • Strongly identifies with 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 user-facing MAX benchmarking, development, and visualization tools and inference APIs

The company offers a next-generation AI developer platform that unifies AI frameworks and hardware, providing unparalleled performance and cost savings. The platform includes a new programming language, Mojo, and a modular engine that enables model deployment, compute portability, and scalability, revolutionizing AI infrastructure.

Company Stage

Series A

Total Funding

$130M

Headquarters

,

Founded

2022

Growth & Insights
Headcount

6 month growth

8%

1 year growth

19%

2 year growth

547%