Senior Software Engineer - AI

Who we are

We're a multidisciplinary team of strategists, designers, engineers and product managers based in London, Delhi, Geneva and Lisbon. Our mission is to help create a world where new digital ideas transform lives - so working here means solving meaningful digital challenges, building your portfolio with standout projects and brands, and using some of the most innovative technologies around.

As a certified B Corporation, honesty and trust are key foundations at Studio Graphene. We believe in using technology as a force for good and, through our work, create a positive impact on the world. We are ISO 9001 and 27001 certified, which demonstrates our commitment to delivering services of the highest quality and maintaining the utmost standards of information security for our clients.

Our culture is supportive, open and collaborative, which has led us to proudly earn a Great Place to Work certification across all our global studios, spanning the UK, Portugal and India. This certification underscores our commitment to nurturing an exceptional workplace culture. We offer great benefits, competitive pay and a chance to make a difference in your career.

What you will do

We are seeking a highly skilled Machine Learning Engineer to design, develop and deploy machine learning models and AI-driven solutions. You will work closely with clients, software engineers and product teams to transform data into actionable insights and intelligent applications.

Responsibilities

• Design and build agentic AI systems - tool use, multi-agent orchestration, ReAct/chain-of-thought pipelines.

• Develop and deploy LLM-powered features: RAG pipelines, autonomous workflow automation.

• Own prompt engineering and context engineering: manage context windows, token budgets, and output structuring.

• Build and maintain backend services using Python and FastAPI; design and manage relational databases with SQLAlchemy and Alembic migrations.

• Host and serve custom models on AWS SageMaker; manage endpoints, scaling, and inference pipelines.

• Work with open-source models via Hugging Face - model selection, inference, and lightweight customisation.

• Set up and maintain ML/AI-Ops workflows - CI/CD pipelines for model deployment, automated testing, and continuous delivery of AI features.

• Stay updated with the latest advancements in ML, AI, deep learning, and agentic frameworks.

Must have

• 4+ years of experience in Python (3.10+) — clean, production-grade code with async patterns and FastAPI.

• GenAI fundamentals: solid understanding of tokens, embeddings, prompt engineering, and context engineering.

• Familiarity with Agent-to-Agent (A2A) and Model Context Protocol (MCP).

• Basic ML/NLP knowledge: classification, regression, NLP pipelines, BERT, LIWC, Bag of Words, and embedding-based similarity models.

• Hands-on experience deploying and hosting custom models on AWS SageMaker.

• Familiarity with the Hugging Face ecosystem and open-source model landscape.

• Experience with ML/AIOps - CI/CD pipelines for model lifecycle, model versioning, monitoring, and automated evaluation in production.

• Strong grasp of data structures, algorithms, and software engineering principles.

• Excellent problem-solving skills with the ability to work independently and collaboratively in a fast-paced environment.

• Good written and verbal communication skills.

Nice to have

• Experience fine-tuning foundation models using PEFT/LoRA techniques.

• Knowledge of guardrails, safety evaluation, or responsible AI principles.

• Monitor models in production - track drift, performance degradation, hallucination rates, token usage, and latency.

What we offer:

• The opportunity to have a real impact on the company's growth and evolution.

• Work for a company with global offices and the opportunity to travel and meet your colleagues.

• Work on a wide variety of tasks that are often ground-up builds.

• Dynamic and fun working environment - we run numerous brown bag sessions, bug-finding days and after-work social events.

• We take care of you and offer multiple perks, including special days off for health & wellness.

• We support you with our sponsored learning & development program.

Studio Graphene is proud to be an equal opportunity workplace. We cultivate a culture of inclusion for all employees, which respects their strengths, views and experiences. We strongly value diversity within our team, recognising that it fosters better decision-making, innovation and ultimately contributes to our business success.