AI Research Engineer (Tool Calling Safety)
AI Research Engineer (Tool Calling Safety)
Employer Active
Job Alert
You will be updated with latest job alerts via email
Share
Job Alert
You will be updated with latest job alerts via email
Job Description
About StackOne
StackOne is the integration gateway for SaaS & AI Agents. Weve raised $24m in 2 years and were backed by top tier investors like Google Ventures and Workday Ventures.
Our customers are established SaaS companies & small to large businesses building AI Agents.
About the role
Were building a world where users of any SaaS can integrate with the tool of their choice in one click thanks to StackOne. Were convinced AI is pivotal to making this a reality by transforming how Software talks to each other and this is what you will help achieve.
Were looking for an AI Research Engineer to secure the next generation of tool-calling for LLMs. Youll research and prototype defences that limit the attack surface of emerging standards such as MCP servers and make tool execution safer against prompt injection and malicious calls.
Youll work with our researchers on state-of-the-art tool-calling models and with our engineers on the execution engine that runs them.
You will own design and run experiments on cutting-edge architectures manage distributed training clusters and help curate & generate high-quality datasets. This role sits closer to the research/ML infra side than product engineering but with a strong mandate for applied production-ready results.
In this role you will work with wider AI team of StackOne (comprising of other researchers and engineers) and report directly to the CTO.
Responsibilities
Investigate vulnerabilities in tool-calling protocols (MCP OpenAPI custom plug-ins).
Design and test mechanisms to restrict malicious or unsafe tool calls.
Collaborate on the architecture of a secure tool execution engine.
Partner with the core research team to align safety work with model quality.
Design and manage synthetic data generation workflows to augment real-world datasets.
Develop tools for dataset curation labeling filtering and augmentation.
Conduct benchmarking and evaluations to measure fine-tuning impact.
Collaborate with the rest of the engineering team to integrate fine-tuned models into production stacks.
Stay ahead of research in parameter-efficient fine-tuning synthetic data and LLM training.
What were looking for
Knowledge of prompt injection sandboxing and least-privilege patterns.
Experience building or auditing APIs plug-ins or protocol runtimes.
Ability to design and evaluate defences for real-time agent systems.
Background in Gen AI with experience with Tool Calling
Interest in security engineering or applied cryptography.
Desire to work in a fast-paced startup taking ownership of projects e2e and bias towards shipping.
Our Stack
You will have a lot of freedom and agency to chose and work with the technologies you love. Our tech stack includes:
Infra & monitoring: AWS (ECS RDS Elasticache Lambda etc.) Cloudflare Tinybird Modal Dagster Langsmith Datadog Sentry
Languages & Frameworks: Typescript () Python NestJS React Svelte
Other day-to-day Tools: Postman Notion Slack Jira Github and more
Benefits
25 days holiday 1 additional day holiday per year of tenure
Participation in the companys employee share options plan
Private health insurance (including dental & optical)
Health fitness and gift card discounts
1000 for your home office set up 500/year top-up
Paid lunch in the office
Annual team offsite to sunny spots (last ones were in Spain and Portugal )
Join one of Europes fastest-growing startups
Work with a veteran team of ex-employees of Google Microsoft Oracle Coinbase JP Morgan and more
Cycle2Work and Electric Cars scheme
Hybrid work set up - typically 2d in the office
Ready to help us change the game for AI Agents Get in touch and lets chat!
We believe diversity drives innovation. We encourage individuals from all backgrounds to apply. As an equal-opportunity employer we celebrate diversity and are committed to creating an inclusive environment for all employees.
Employment Type
Full-Time
