Overview

Key is reimagining how professionals connect, create, and grow together.

Responsibilities

• 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

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
• Excellent written and verbal communication in English
• Languages and frameworks: Python, Node.js, PyTorch or TensorFlow, LangChain
• Infrastructure: GCP, AWS, Docker, Kubernetes
• Data and ML: PostgreSQL, vector dat .