About this role
AI Library Architect at Modular. Location: Los Altos or Seattle or United States or Canada or United Kingdom or Europe. Role: designing APIs, building implementations, writing specs Requirements: Significant experience designing developer-facing APIs and SDKs; deep understanding of modern AI frameworks (PyTorch, JAX, TensorFlow, vLLM, XLA/MLIR); experience with inference and training systems; fluency in C++, Rust, Go and Python; strong spec-writing and API ergonomics. Category: Engineering Seniority: Senior Level Tools: MAX Framework, PyTorch, JAX, TensorFlow, vLLM, XLA, MLIR, C++, Rust, Go, Python, Mojo Commitment: Full Time Workplace: Hybrid Languages: English