AI Platform Engineer
AHEAD
IndiaremotePosted 17 days ago
Skill Required
AI-Platform-EngineeringMachine-Learning-EngineeringMLOpsPlatform-EngineeringDevOpsAI-Platform-EngineerPlatform-AI-EngineerAI-ML-Platform-EngineerAI-Agent-Platform-EngineerAI-Data-Platform-EngineerMachine-Learning-Platform-EngineerAI EngineerPlatform EngineerAIMachine LearningTensorFlowKubernetesautomationObservabilityTerraformanalyticsPyTorchbuildingMLflowJupyterAnsiblePythondesignAzureCI/CDCloudDesign PatternsAWSGCPandCICDFulltime
Key highlights
- 4+ years platform architecture experience (2+ years focused on AI/ML).
- Strong proficiency in Python and experience with TensorFlow or PyTorch.
- Certification in Kubernetes or major cloud platforms (AWS, Azure, GCP).
- Comprehensive health insurance coverage for employees and dependents.
- Flexible work arrangements supporting work‑life balance.
- Experience with high‑performance computing (HPC) environments.
Role overview
AHEAD builds platforms for digital business by weaving together advances in cloud infrastructure, automation, analytics, and software delivery to help enterprises achieve digital transformation. The company is seeking an experienced AI Platform Engineer to design, deploy, and optimize AI/ML infrastructure, AI workflows, and automated pipelines, focusing on building scalable environments for training and deploying machine learning models using modern orchestration, automation, and GPU acceleration technologies.
Responsibilities
- Architect and manage Kubernetes clusters tailored to AI/ML workloads.
- Implement Run:ai and operators for GPU resource orchestration and workload scheduling.
- Develop and maintain Python-based automation scripts and ML pipelines; automate infrastructure provisioning with Terraform and configuration management with Ansible.
- Create and manage Jupyter Notebooks for experimentation and collaboration.
- Integrate and optimize NVIDIA Enterprise Suite components (CUDA, NeMo Framework, Triton, TensorRT, GPU drivers) for accelerated computing.
- Establish and maintain MLOps best practices for model lifecycle management, CI/CD, and monitoring (e.g., MLflow, Kubeflow).
- Work closely with data scientists and platform engineers to ensure efficient resource utilization and scalability across environments.
Requirements
- 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.
- 4+ years in platform architecture or solutions architecture, with 2+ years focused on AI/ML workloads.
- Experience with high‑performance computing (HPC) environments.
- Familiarity with distributed training and model optimization techniques.
- Certification in Kubernetes or cloud platforms (AWS, Azure, GCP).
Benefits
- 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.
Additional details
- At AHEAD, we prioritize creating a culture of belonging, where all perspectives and voices are represented, valued, respected, and heard.
- We create spaces to empower everyone to speak up, make change, and drive the culture at AHEAD.
- We are an equal opportunity employer and do not discriminate based on protected characteristics.
- We embrace all candidates that will contribute to the diversification and enrichment of ideas and perspectives at AHEAD.
- Through our daily work and internal groups like Moving Women AHEAD and RISE AHEAD, we value and benefit from diversity of people, ideas, experience, and everything in between.
- We fuel growth by stacking our office with top‑notch technologies in a multi‑million‑dollar lab, encouraging cross‑department training and development, and sponsoring certifications and credentials for continued learning.
- Originally posted on Himalayas.