Applied Ai Engineer Ahmedabad (India)

About Us

Key is reimagining how professionals connect, create, and grow together. We're building a platform that transforms skilled networking by focusing on authentic relationships and meaningful connections. We believe in the power of real human connection, amplified by technology but driven by people.

The Role

Were looking for an Applied AI Engineer to take AI and ML models from concept to production. You'll build scalable, reliable systems using Python, cloud services, and modern MLOps practices, and own end-to-end delivery of AI solutions.

Responsibilities

End-to-end ownership of production ML systems from prototype to live services

Implement, fine-tune, and optimize models, including LLMs for search and recommendation, for performance, latency, and cost

Design, build, and maintain data and inference pipelines using MLOps principles for CI/CD, monitoring, and automated retraining

Collaborate with backend and frontend to integrate AI into product experiences

Design and maintain well-documented, scalable APIs and services to expose AI capabilities

Proactively improve AI infrastructure, model performance, and incorporate current techniques

Uphold quality via code reviews, testing, and documentation

Qualifications

Bachelors or Masters in Computer Science, Engineering, or a related field

4+ years of professional software engineering experience, including 2+ years deploying ML systems to production

Expertise in Python and ML libraries such as PyTorch, TensorFlow, and scikit-learn

Robust experience with AWS, GCP, or Azure and their ML services (for example, SageMaker or Vertex AI)

Solid understanding of API design and system architecture

Experience with LLMs, RAG systems, and vector databases is a robust plus

Excellent written and verbal communication in English

Tech Stack

Languages and frameworks: Python, Node.js, PyTorch or TensorFlow, LangChain

Infrastructure: GCP, AWS, Docker, Kubernetes

Data and ML: PostgreSQL, vector dat .