Home

Sr. ML Compiler Engineer - Automated Reasoning Science, Annapurna Labs at Amazon Web Services (AWS)

Location: Cupertino, CA

Job Summary:

Job Duties:
- Develop and optimize deep learning compiler stack
- Optimize application models from frameworks like PyTorch, TensorFlow, and JAX
- Collaborate with cross-functional teams

Required Skills (Keywords):
- Compiler design
- Deep learning
- Optimization techniques
- Python/C++
- Knowledge of machine learning frameworks

Required Experiences (Topics):
- Experience in compiler engineering
- Background in deep learning models
- Familiarity with large-scale applications
- Prior work with optimization algorithms

Job URLs: