Contribute to the definition of AI platform for LLMs, develop AI back-end for de-facto standard AI frameworks.

What you'll be doing:
Understand, analyze, profile, and optimize LLM workloads for AI platform

Collaborate with researchers and engineers in the brand and externally, providing insights useful for HW design and work on improving of the performance of third party workloads

Design and implement production-quality AI framework backend
Understand and solve challenges pertaining to a hybrid AI platform

Preferred Experience
Experience with enabling and optimization of training or/and inference of specific LLMs with more than B parameters on target HW
Familiarity with Deep Learning and Transformers/LLMs from open source repos like Hugging Face Hub
Experience with modern AI frameworks (Tensorflow, Pytorch) - 3+ years
Experience with Python and C: 5+ years
Familiarity with DeepSpeed, Horovod or other distri...