Lead ML Engineer - Generative AI
Apple
- Hyderabad, Telangana
- Permanent
- Full-time
- Solid understanding and experience in advanced machine learning algorithms, deep learning, statistics, and large language models.
- Demonstrable experience with transformer models such as BERT, GPT, RoBERTa, etc, and a solid understanding of their underlying principles is a plus
- Model optimization and fine tuning of large language models for better performance, latency, memory and throughput
- Strong software development skills with proficiency in Python, PyTorch, TensorFlow, JAX
- Building MLOps infrastructure for experimentation and A/B testing and production deployment and operations.
- Experience with MLOps technology: Git, Docker, Kubernetes, MLFlow/DVC/ClearML, Airflow
- Setting up LangChain pipeline including prompt engineering and parsing
- Ability to meaningfully present results of analyses in a clear and impactful manner. Able to break down sophisticated machine learning model knowledges to digestible information for non-technical audience.
- Experience in applying ML techniques in developer productivity is a plus
- 5+ years of overall experience in the software industry focussed on AI/ML