10 Sep
|
CriticalRiver
|
Hyderabad
10 Sep
CriticalRiver
Hyderabad
Job Summary
We are seeking a Senior AI / ML Architect with 10–15 years of experience and robust hands-on expertise in machine learning, deep learning, LLM model training, domain-specific fine-tuning, cloud deployment, and model hosting.
The primary focus of this role will be to train and fine-tune open-weight AI/ML models using domain-specific enterprise data and build scalable infrastructure to host, serve, manage, and monitor these models in production.
This is a highly hands-on role requiring strong practical experience with PyTorch, Hugging Face, LoRA, QLoRA, GPU infrastructure, cloud platforms, model serving, and MLOps tools. The candidate should be an experienced technical architect who can make architecture decisions while remaining deeply involved in implementation and troubleshooting.
Key Responsibilities
Architect and implement solutions for training and fine-tuning ML and LLM models using domain-specific enterprise data.
Evaluate and select appropriate open-weight models based on domain requirements, model capability, size, performance, licensing,
infrastructure requirements, and cost.
Build end-to-end pipelines for training data preparation, model training, fine-tuning, evaluation, model packaging, deployment, and monitoring.
Perform hands-on LoRA and QLoRA fine-tuning of open-weight models.
Configure and optimize fine-tuning parameters including rank, alpha, target modules, learning rate, batch size, sequence length, quantization, and training strategy.
Work with domain-specific datasets to perform data cleaning, transformation, instruction formatting, dataset curation, synthetic data generation, deduplication, and quality validation.
Train and fine-tune models using PyTorch, Hugging Face Transformers, PEFT, TRL, Accelerate, and related frameworks.
Build scalable GPU-based training environments on AWS, Azure, or GCP.
Plan and optimize GPU infrastructure for model training and inference, including A100, H100, L40S, or equivalent GPU environments.
Deploy
📌 Senior AI (Hyderabad)
🏢 CriticalRiver
📍 Hyderabad