Job Title

Staff ML Engineer (Level 5) - Location: India | Type: Full-Time. Team: Gaia Emblem Engine.

Job Summary

Cohesity is looking for a Staff ML Engineer (Level 5) to help design and scale the Gaia Emblem engine, the intelligence layer powering our next-generation data platform. In this role, you will architect and build large-scale, production-grade ML/LLM systems - spanning model serving, retrieval-augmented generation (RAG), GPU infrastructure, and distributed backend services. As a Staff-level engineer, you will set technical direction, mentor senior engineers, and partner closely with Product and Data Engineering to bring Gaia Emblem's roadmap to life.

Key Responsibilities

- Architect and drive the technical roadmap for the Gaia Emblem engine, including LLM integration, RAG pipelines, and prompt-engineering frameworks.
- Design and operate scalable, GPU-aware infrastructure on Kubernetes (vanilla K8s, OpenShift, EKS, GKE, or equivalent), including GPU scheduling, autoscaling, and costefficient resource utilization.
- Build resilient, high-throughput distributed systems and microservices exposed via gRPC and REST APIs.
- Own data engineering pipelines feeding ML workflows, including ingestion, transformation, and storage across SQL, NoSQL, and search systems (PostgreSQL, Redis, Elasticsearch).
- Define and enforce engineering best practices: CI/CD, test automation, observability, and infrastructure-as-code (Terraform, Helm).
- Lead design reviews, set coding and architectural standards, and mentor engineers across the organization.
- Partner with Product, Applied ML,



and SRE teams to translate business requirements into scalable technical solutions.
- Drive root-cause debugging and performance optimization across distributed, GPUbacked services.
- Evaluate and introduce new tools, frameworks, and infrastructure patterns to keep Gaia Emblem at the forefront of applied ML engineering.

Minimum Qualifications

Experience

- 10+ years of qualified software engineering experience, including 8+ years building distributed, production-grade backend systems.
- 3+ years of hands-on experience with LLM-based systems, RAG architectures, or applied ML infrastructure.
- Demonstrated experience operating services on Kubernetes (vanilla K8s, OpenShift, EKS, GKE, or similar) at production scale, including GPU-backed workloads.
- Track record of technical leadership: driving architecture decisions, leading cross-team initiatives, and mentoring engineers at Staff/Senior level.

Education

- Bachelor's degree in Computer Science, Engineering, or a related technical field required.
- Master's degree or PhD in Computer Science, Machine Learning, or a related field preferred.

Required Skills

- Languages: Python, Java, Go, JavaScript
- ML/AI: LLMs, Prompt Engineering, Retrieval-Augmented Generation (RAG)
- Infrastructure Cloud:



Kubernetes (platform-agnostic - vanilla K8s, OpenShift, EKS, GKE, or equivalent), GPU Scheduling, Docker, Helm Charts, Terraform, AWS, Cloud Computing
- Distributed Systems APIs: Distributed Systems, Microservices, gRPC, REST APIs
- Data Storage: Data Engineering, SQL, NoSQL, PostgreSQL, Redis, Elasticsearch, Databases
- Engineering Practices: CI/CD, DevOps, Test Automation, Git, Debugging, Object-Oriented Programming (OOP)

Preferred / Nice-to-Have Skills

- Node.js, Back-End Web Development
- Jenkins and other CI/CD tooling
- Experience with formal verification or correctness-focused engineering practices - Prior experience in enterprise data protection, storage, or infrastructure software

Expectations from the Role

- Operate with Staff-level ownership: define the what and how for a major component of Gaia Emblem, not just execute tickets.
- Balance long-term architectural vision with pragmatic, incremental delivery.
- Act as a force multiplier - raise the technical bar of the team through design reviews, documentation, and mentorship.
- Communicate complex technical trade-offs clearly to both engineering and nonengineering stakeholders.
- Be hands-on: this role is expected to write and review production code regularly, not purely architect from a distance.

Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.

📌 Staff ML Engineer (Pune)
🏢 Cohesity
📍 Pune