Applied AI Engineer

About Us

Key is reimagining how professionals connect, create, and grow together. We're building a platform that transforms professional 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 new 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
• Strong 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 strong 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 databases such as Pinecone or pgvector, MLflow, Airflow
• Tooling: Git and CI/CD systems

What we value

• User obsession: Every model we build must enhance user experience and deliver measurable value.
• Excellence in execution: We prioritize clarity, precision, and reliability in everything we ship.
• Continuous learning: We stay curious, challenge assumptions, and adapt rapidly to emerging technology.
• Collaboration and ownership: Great products are built through shared ownership and mutual respect.

What We Offer

• Competitive salary
• Flexible PTO policy
• Regular team offsites and events
• Leadership development opportunities while staying technical
• The autonomy to shape both product experiences and technical direction

We're committed to building a diverse and inclusive team. We encourage applications from people of all backgrounds, identities, and experiences.

Location

Ahmedabad, Gujarat, India