The AI Platform Engineer will design and optimize AI/ML infrastructure, automate pipelines, and manage Kubernetes for effective machine learning model deployment and collaboration.
AHEAD builds platforms for digital business. By weaving together advances in cloud infrastructure, automation and analytics, and software delivery, we help enterprises deliver on the promise of digital transformation.
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 an individual's race, national origin, color, gender, gender identity, gender expression, sexual orientation, religion, age, disability, marital status, or any other protected characteristic under applicable law, whether actual or perceived.
We embrace all candidates that will contribute to the diversification and enrichment of ideas and perspectives at AHEAD.
We are 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 machine learning models, leveraging modern 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.
Key Responsibilities
- Kubernetes for AI/ML: Architect and manage Kubernetes clusters tailored to AI/ML workloads.
- GPU Orchestration: Implement Run:ai and operators for GPU resource orchestration and workload scheduling.
- Automation & Pipelines: Develop and maintain Python-based automation scripts and ML pipelines; automate infrastructure provisioning with Terraform and configuration management with Ansible.
- Notebooks & Collaboration: Create and manage Jupyter Notebooks for experimentation and collaboration.
- NVIDIA Integration: Integrate and optimize NVIDIA Enterprise Suite components (CUDA, NeMo Framework, Triton, TensorRT, GPU drivers) for accelerated computing.
- MLOps Practices: Establish and maintain MLOps best practices for model lifecycle management, CI/CD, and monitoring (e.g., MLflow, Kubeflow).
- Collaboration: Work closely with data scientists and platform engineers to ensure efficient resource utilization and scalability across environments.
Required Skills & Experience
- 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.
Qualifications
- 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).
Why 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, by encouraging cross department training and development, sponsoring certifications and credentials for continued learning.
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
AHEAD Gurugram, Haryana, IND Office
One Horizon Center Golf Course Road, DLF Phase V Sector 43, Gurugram, India, 122002
Similar Jobs
.png)
Cloud • Information Technology
Leads customer-facing design and implementation of cloud, platform, infrastructure, data, and AI solutions. Responsibilities include cloud migration and modernization, Infrastructure as Code, CI/CD, Kubernetes, automation, GitOps, platform security, observability, and AI-assisted development. The consultant conducts workshops, creates reusable platform capabilities, enables responsible AI adoption, mentors engineers, supports proposals and pre-sales, and ensures secure, reliable, governed, and cost-effective production solutions.
Top Skills:
Amazon BedrockAmazon Q DeveloperAPIsAWSAws MgnAws Migration HubAzureAzure MigrateAzure OpenaiBashBicepCi/CdClaude CodeCrewaiCursorDevinFinopsGenerative AiGithub CopilotGitopsGCPGoogle Migration CenterGoogle Vertex AiInfrastructure As CodeKubernetesLangchainLanggraphLlamaindexLlmsMicrosoft FoundryModel ApisObservabilityPolicy As CodePowershellPrompt EngineeringPythonRagSagemakerSemantic KernelService MeshTerraformVector DatabasesVmware HcxWindsurf
AdTech • Marketing Tech
Maintain and optimize production LLM/VLM services and microservices (Flask/FastAPI), manage async queues, deploy and monitor Uvicorn/Gunicorn hosts, integrate LLM routing tools, design prompt strategies, build feedback/evaluation pipelines, expose secure REST APIs, and track token usage, latency, and errors to ensure SaaS-grade AI performance.
Top Skills:
AWSCeleryCi/CdDockerDramatiqFastapiFine-TuningFlaskGCPGunicornLlmsLogging/MonitoringOpenrouterPrompt EngineeringPythonRest ApiRqUvicornVlms
Healthtech • Pharmaceutical • Manufacturing
Lead end-to-end delivery of Power Platform and AI-enabled automations using Power Apps, Power Automate, Copilot Studio and PowerShell. Implement and govern Copilot/agent solutions, troubleshoot production incidents, support platform workloads, maintain operational runbooks, and build Power BI dashboards for automation health, usage, and governance metrics.
Top Skills:
Azure Ai FoundryCopilot StudioDataverseEntra IdExchange OnlineLlm-Based AutomationsMicrosoft 365Microsoft Graph ApiMicrosoft TeamsPower AppsPower AutomatePower BIPowershellSharepoint Online
What you need to know about the Delhi Tech Scene
Delhi, India's capital city, is a place where tradition and progress co-exist. While Old Delhi is known for its rich history and bustling markets, New Delhi is defined by its modern architecture. It's clear the region places a strong emphasis on preserving its cultural heritage while embracing technological advancements, particularly in artificial intelligence, which plays a central role in shaping the city's tech landscape, fueled by investments in research and development.
