Home

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

Location: Cupertino, CA

Job Summary:

Job Duties:
- Develop a deep learning compiler stack for optimization
- Collaborate with teams to enhance application models
- Work with frameworks like PyTorch, TensorFlow, and JAX
- Contribute to compiler design and implementation

Required Skills (Keywords):
- Compiler engineering
- Deep learning optimization
- PyTorch
- TensorFlow
- JAX
- Programming languages (e.g., C++, Python)

Required Experiences (Topics):
- Compiler development
- Deep learning frameworks
- Optimization techniques
- Team collaboration and project delivery

Job URLs: