AI Performance Optimization Team Lead
Benefits:
Medical, dental and vision
Life and AD&D insurance
Flexible paid time off
Paid family leave benefits
$500 monthly meal reimbursement
Education Requirements:
Bachelor’s degree in Computer Science, Engineering, or a related field
Advanced degree (Master’s or PhD) in machine learning, compilers, or systems highly preferred
Experience Requirements:
Strong expertise with deep learning frameworks (PyTorch, JAX, or TensorFlow)
Experience profiling models on hardware accelerators
Experience with model optimization techniques
Deep understanding of compiler internals
Experience with CUDA, Triton, or GPU programming models
Other Requirements:
Knowledge of distributed computing and parallelism strategies
Familiarity with software engineering practices (APIs, CI/CD)
Track record contributing to open-source projects
Strategic leadership for revenue-driving functions
Responsibilities:
Own technical direction of performance-oriented model optimization efforts
Advance compiler technology through optimization passes and graph transformations
Ensure optimizations are accessible via clean APIs and automated tooling
Design profiling and debugging tools to identify bottlenecks
Service high impact customer deals as key technical and strategic leader
Show more details