Staff Software Engineer (AI)

Location: Pune, Maharashtra, India

Department: Developer Experience (DX)

About Druva
Druva is the resilience foundation for the AI enterprise, helping organizations secure and recover from connected risk across data, cyber, identity, and AI. The Resilience Cloud is a fully managed, cloud-native SaaS platform that delivers air-gapped and immutable protection across cloud, SaaS, on-premises, endpoint, and edge environments. Powered by Dru MetaGraph, Druva’s graph-powered intelligence layer, the platform connects critical business context so customers can understand risk, respond faster, recover cleanly, and govern data with greater confidence.

Trusted by nearly 7,500 customers, including 75 of the Fortune 500, Druva helps safeguard the critical information and systems businesses depend on in an increasingly connected world.

Visit druva.com and follow us on LinkedIn, X and Facebook.

Experience Required: 3 to 5 years
Location: Pune
Employment Type: Full-time

About the Role

As a Staff Software Engineer (AI), you will lead the design and development of cutting-edge AI solutions. This role involves taking ownership of projects, mentoring junior engineers, and driving innovation in AI systems. You will work closely with stakeholders to build scalable and robust AI applications that solve complex problems.

Key Responsibilities

• Lead the design, development, and deployment of scalable AI/ML systems.
• Drive technical discussions and decisions related to AI architecture, algorithms, and tools.
• Collaborate with data scientists, engineers, and product teams to align AI solutions with business goals.
• Own the end-to-end lifecycle of AI models, from research and development to production and monitoring.
• Implement MLOps pipelines for model training, validation, deployment, and monitoring.
• Optimize models for performance, accuracy, and scalability in production environments.
• Mentor and guide junior engineers to build their technical skills and understanding of AI concepts.
• Research and experiment with new algorithms and technologies to improve existing systems.

Required Skills & Qualifications

• Education: Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or related disciplines.
• Programming Languages: Advanced expertise in Python and familiarity with other languages such as Java or C++.
• AI/ML Expertise: In-depth knowledge of machine learning, deep learning, NLP, or computer vision.
• Hands-on experience with ML frameworks such as TensorFlow, PyTorch, or Hugging Face.
• Strong understanding of distributed systems, cloud computing, and microservices architecture.
• Proficiency in deploying AI/ML systems on cloud platforms (AWS, GCP, or Azure).
• Experience with big data tools and frameworks (Spark, Hadoop, or similar).
• Demonstrated ability to lead technical teams and projects.

Preferred Skills

• Experience with advanced AI concepts like federated learning, GANs, or transfer learning.
• Familiarity with data engineering tools and pipelines.
• Expertise in containerization (Docker) and orchestration (Kubernetes).
• Strong problem-solving, leadership, and communication skills.