Location: Austin, TX
Job Summary:
Job Duties:
- Develop and optimize a deep learning compiler stack.
- Collaborate with teams on application models for various domains.
- Improve performance and efficiency of models from frameworks like PyTorch, TensorFlow, and JAX.
Required Skills:
- Compiler design
- Deep learning
- Optimizations
- Knowledge of PyTorch, TensorFlow, JAX
- Programming languages (e.g., C++, Python)
Required Experiences:
- Compiler engineering
- Deep learning frameworks
- Performance tuning techniques
- Software development in relevant environments
Job URLs: