1 Pytorch Distributed Jobs

Setup a job Alert
JobPe aggregates results for easy application access, but you actually apply on the job portal directly.

gnani.ai logo
Senior ML Engineer gnani.ai

5.0 - 9.0 years

0 Lacs

karnataka

On-site

IndiaAI is currently focused on developing India's next-gen foundational LLMs and is seeking a Senior ML Engineer who is skilled in large-scale pre-training, distributed GPU systems, and data creation pipelines. As a Senior ML Engineer, your primary responsibility will be to work with cutting-edge technologies such as Megatron-LM, NVIDIA NeMo, DeepSpeed, PyTorch Distributed, and SLURM to train models exceeding 7B parameters on multi-node GPU clusters. Key Responsibilities: - Build and optimize LLM pre-training pipelines for models exceeding 7B parameters. - Implement distributed training using PyTorch Distributed, DeepSpeed (ZeRO/FSDP), Megatron-LM, and NVIDIA NeMo. - Manage multi-node GPU j...

Posted 2 days ago

AI Match Score
Apply
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

Google Play App Store

Featured Companies

View All Companies