Home

Compiler Engineer II - Machine Learning, Annapurna Labs at Amazon Web Services (AWS)

Location: Austin, TX, US

Job Summary:

Job Duties and Scopes:
- Design, develop, and optimize features for a deep learning compiler stack.
- Collaborate with cross-functional teams for system-wide performance optimization.
- Address technical challenges in compiler technology and deep learning software.
- Participate in design discussions and code reviews.

Required Skills:
- Proficiency in programming languages, primarily C++ and Python.
- Knowledge of compiler design, instruction scheduling, memory allocation, and parallel programming.
- Experience with tools like LLVM and MLIR.

Required Experiences:
- Minimum 3 years of software development experience.
- At least 2 years of experience in compiler optimization or related fields.
- Experience with TensorFlow, PyTorch, or JAX, and understanding of deep-learning models.

Job URLs: