Job Description:Day-to-day work involves working on Frameworks such as PyTorch, Tensorflow, JAX etc. The job role involves design
and developing features in AI frameworks software stack. You will be participating in enabling and optimizing state of
The roles and responsibilities that you would need to performance may include the following:
- Design and develop SW features for AI frameworks - both HW-agnostic and HW-aware.
- Enhance and extend the Deep learning training, and Inference capabilities in the Software stack
- Identifying optimization opportunities in the software stack to enhance performance of Deep learning workloads
- Participate in discussions with Open-source community, involve in development and adopting open source features in our SW stack.
Experience:
- BTech, MS or PhD in CS or related fields with an overall experience of 8 to 15 years.
- Should have experience in leading/managing a group of engineers for at least 4 years.
- Proficient in Advanced C++ (C++ 14/17) and Intermediate skills of Python and parallel programming.
- In depth & hands on experience in one of the frameworks such as PyTorch, Tensorflow or JAX
- Ability to debug complex issues in multi layered SW systems. Understanding of SW integration in large open-source frameworks.
- Strong understanding of computer architecture and HW-SW optimization techniques.
- Experience working on frameworks/platforms that have gone to production.
- Effective communication skills and experience with working in a cross-geo teams.
Preferable
- Knowledge of compiler algorithms for heterogeneous system or triton backend.
- Hands-on experience of Math kernel operators in Pytorch or Tensorflow
- Practical knowledge of Deep Learning models/LLMs for Vision / NLP
Experienced HireShift 1 (India)India, Bangalore