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