Staff Engineer, AI/ML

About the role

Checkout.com is looking for a Staff Engineer in AI/ML to drive the company’s AI vision and bring large‑scale AI innovations to production. You will work within the Data and AI platform team, pioneering real‑world AI/ML solutions, MLOps, and large language models across the business.

Key responsibilities

• Collaborate with cross‑functional teams to research, scope, and validate AI use cases that deliver business value.
• Drive AI adoption by rigorously evaluating ideas, championing high‑impact applications, and pushing back on unsuitable use cases.
• Design, refine and build the MLOps components of the data and AI platform, including vector databases, feature stores and model serving at millisecond latency.
• Implement CI/CD pipelines and enforce best practices for model deployment, security and regulatory compliance.
• Participate in the AI/ML guild, influencing new approaches and use‑case development.
• Continuously monitor and optimise system performance for scalability, security and operational efficiency.

Required profile

• Proficiency in Python (plus at least one additional language) and experience with libraries such as PyTorch, Pandas and Hugging Face Transformers.
• Working knowledge of common AI/ML models and their application to solve specific problems.
• Strong engineering background in designing and implementing services, data models and features.
• Expertise with cloud platforms (AWS, GCP or Azure) and containerisation tools (Docker, Kubernetes).
• Experience with modern data platforms (BigQuery, Databricks) and data‑processing workflows (ETL, pipelines).
• Hands‑on experience with cloud‑hosted AI services (Bedrock, SageMaker, VertexAI).
• Excellent problem‑solving ability and capacity to learn quickly in a fast‑paced environment.
• Strong communication skills and ability to work effectively across diverse teams.

Required skills

• Python
• PyTorch
• Pandas
• Hugging Face Transformers
• AWS
• GCP
• Azure
• Docker
• Kubernetes
• BigQuery
• Databricks
• Bedrock
• SageMaker
• VertexAI

What we offer

• Hybrid working model with three days onsite per week.
• Office snacks, breakfast and lunch options at all locations.