About this role
About the Role AHEAD is seeking an experienced AI Platform Engineer to design, deploy, and optimize AI/ML infrastructure, AI workflows, and automated pipelines. This role focuses on building scalable environments for training and deploying ML models, leveraging orchestration, automation, and GPU acceleration technologies. You will collaborate with data scientists and platform engineers to drive efficient resource utilization and scalable operations across cloud and hybrid environments. What You'll Do
- Design, deploy, and optimize AI/ML infrastructure, workflows, and automated pipelines.
- Build scalable environments for training and deploying ML models; leverage orchestration, automation, and GPU acceleration.
- Collaborate with data scientists and platform engineers to drive resource utilization and scalable operations across cloud/hybrid environments. What We're Looking For
- Strong proficiency in Python and experience with ML frameworks (TensorFlow, PyTorch).
- Hands-on experience with Kubernetes and container orchestration.
- Familiarity with Run:ai or similar GPU scheduling platforms.
- Expertise in Terraform and Ansible for infrastructure automation.
- Experience with Jupyter Notebooks for ML development.
- Knowledge of NVIDIA Enterprise Suite (CUDA, NeMo Framework, Triton, GPU drivers).
- Solid understanding of MLOps principles and tools (e.g., MLflow, Kubeflow).
- Background in deploying and scaling AI workloads in cloud or hybrid environments. Compensation & Benefits
- India Employment Benefits include: Comprehensive health insurance coverage for employees, with options to extend coverage to dependents; Paid time off and company holidays, along with additional leave benefits as per policy; Flexible work arrangements, supporting work-life balance; Learning and development opportunities to support continuous growth and upskilling; Employee wellness initiatives and programs focused on physical and mental well-being; Retirement and statutory benefits in line with India regulations; Inclusive and people-first culture, with a strong focus on collaboration and ownership.