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
- Solid 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 dat
📌 Applied AI Engineer (Ahmedabad)
🏢 Key
📍 Ahmedabad