Sr AI Engineer (RAG Specialist with Strong Python Skills)

About the role

We are seeking a highly skilled AI Engineer with expertise in Retrieval-Augmented Generation (RAG) and strong Python programming skills to join our innovative team.

Key Responsibilities

• Test, evaluate and implement AI models with a focus on Retrieval-Augmented Generation (RAG).
• Strong prompt engineering and prompt testing skills to derive the desired outcome.
• Collaborate with cross-functional teams to integrate AI solutions into existing systems.
• Optimize and fine-tune RAG models for performance and scalability cost saving.
• Conduct research and stay up-to-date with the latest advancements in AI and machine learning.
• Design, build, and deploy agentic AI systems capable of autonomous planning, tool use, and multi-step task execution using frameworks such as LangGraph, AutoGen, or CrewAI.
• Architect and deploy AI/ML and RAG solutions on AWS, leveraging services such as SageMaker, Bedrock, Lambda, and S3.
• Write clean, efficient, and maintainable code in Python.
• Perform data preprocessing, feature engineering, and model evaluation.
• Troubleshoot and debug AI models and applications in a mono-repo settings.
• Document AI models, processes, and workflows.

Qualifications

• Proven experience as an AI Engineer or similar role.
• Strong proficiency in Python programming.
• In-depth knowledge of Retrieval-Augmented Generation (RAG) models.
• Proven experience in prompt engineering.
• Experience with machine learning frameworks and libraries (e.g., TensorFlow, PyTorch).
• Proficiency in data preprocessing and feature engineering techniques.
• Hands-on experience building agentic AI applications, including tool/function calling, multi-agent orchestration, and autonomous workflow design.
• Demonstrated experience with AWS cloud services (e.g., SageMaker, Bedrock, Lambda, EC2, S3) for developing and deploying AI/ML solutions.
• Excellent problem-solving skills and attention to detail.
• Strong communication and teamwork abilities.

Preferred Skills

• Experience with cloud platforms, particularly AWS (e.g., SageMaker, Bedrock, Lambda, ECS/EKS, S3); familiarity with Azure or Google Cloud is a plus.
• Experience with agentic AI frameworks and tooling (e.g., LangGraph, LangChain, AutoGen, CrewAI, AWS Bedrock Agents, Model Context Protocol).
• AWS certification (e.g., AWS Certified Machine Learning – Specialty or AWS Certified Solutions Architect).
• Knowledge of natural language processing (NLP) techniques.
• Familiarity with version control systems (e.g., Git).
• Experience with deploying AI models in production environments.

Benefits

We offer a flexible working pattern, including working flexible hours to help you fit everything in and work when you are the most productive. We also provide a range of country-specific benefits, including wellness initiatives, shared parental leave, study assistance, and sabbaticals.

Pay

U.S. National Base Pay Range: $104,900 - $174,700. Geographic differentials may apply in some locations to better reflect local market rates.

Schedule

This job is eligible for an annual incentive bonus.