remote
Senior AI Engineer - Amplitude
AI Engineer
Senior AI Engineer responsible for designing, building, and scaling AI agents that power analytics features, leveraging Python, deep learning frameworks, and cloud services to deliver real‑time insights for product and marketing teams.
About the role
Key Responsibilities
- Design and implement end‑to‑end AI models and agents that automate data analysis, user behavior prediction, and recommendation generation.
- Collaborate with product, data, and engineering teams to translate business requirements into scalable machine‑learning solutions.
- Develop, train, and fine‑tune deep‑learning models using TensorFlow or PyTorch, ensuring high accuracy and low latency in production.
- Deploy and monitor AI services on AWS, managing containerized workloads, CI/CD pipelines, and performance metrics.
- Lead code reviews, mentor junior engineers, and champion best practices for model versioning, reproducibility, and ethical AI.
Requirements
- 5+ years of professional experience in machine learning or AI engineering, with a strong focus on production‑grade systems.
- Proficiency in Python and deep‑learning frameworks such as TensorFlow or PyTorch.
- Hands‑on experience deploying scalable AI solutions on AWS (e.g., SageMaker, ECS/EKS, Lambda).
- Solid understanding of data pipelines, feature engineering, and model evaluation techniques.
- Excellent problem‑solving skills, ability to work cross‑functionally, and a growth‑mindset orientation.
Skills
pythonmachine learningdeep learningtensorflowpytorchaws