Modular is hiring a
AI Performance & Runtime Engineer

Summary

Join Modular as an AI Performance Engineer to optimize the performance of Modular's MAX Engine, collaborate with engineers to push cutting-edge performance, and work on automated performance analysis and benchmarking.

Requirements

• 2+ years of professional or postgraduate academic experience working on or researching performance analysis, tooling or benchmarking
• In-depth knowledge of Python
• Experience in C++ programming and complex software systems
• Experience with performance modeling and analysis
• Creativity and curiosity for solving complex problems
• Experience writing production-quality software
• Strong written and verbal communication skills

Responsibilities

• Collaborate with Tooling & Infrastructure teams to design systems for automated performance analysis and benchmarking
• Collaborate with Graph Compiler & Kernels teams to design, triage, and track high-priority in-flight model optimizations
• Work with the product team to understand customer use cases, relevant metrics, and design representative benchmarking pipelines
• Design & develop automated regression detection & reversion system for wide variety of key customer metrics
• Build statistical models and tools to operate on benchmarking & telemetry data and help develop key insights for performance, cloud costs, etc

Preferred Qualifications

• Experience with a major machine learning framework such as TensorFlow, PyTorch, or ONNX
• Experience with cloud computing
• Experience with database systems and statistical modeling

Benefits

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