Senior ML Engineer - AI Labs
MumbaiFull-time4–7 yrsPosted 3 weeks ago
Mid-level data science role at IDFC FIRST Bank in Mumbai. Needs 4+ years and a BCA, B.Tech or B.E..
About the role
As a Senior Machine Learning Engineer within the Data & Analytics team, you will be responsible for managing and optimizing the training infrastructure for Large Language Models (LLMs). This role demands a deep understanding of GPU architecture, machine learning principles, and distributed computing.
What you'll do
- Lead Generative AI projects in a cross-functional team environment
- Apply advanced machine learning principles and algorithms, particularly for LLMs such as GPT-4, BERT, and Transformers
- Utilize deep learning frameworks like TensorFlow, PyTorch, and Keras for model training
- Manage and optimize cloud-based resources (AWS, Azure, GCP) for deep learning model training
- Implement containerization and orchestration using Docker and Kubernetes
What you need
- Machine learning
- LLMs
- Deep learning
- TensorFlow
- MLOps
- PyTorch
Good to have
- Hadoop
- Spark
- Docker
- Kubernetes
- AWS
- GCP