AI/ML Engineer
Join our backed startup in Stockholm to design and evaluate AI features - from RAG to agentic patterns - and see your work powering real applications as we enter our next chapter of growth.
About us
We’re building something big in the future-of-work space, with the talent, technology, and backing to move fast and aim high. We work in tight cycles, give honest feedback, and trust each other to own big, complex problems. Joining now means becoming part of the core team at the very beginning - with the chance to shape both the product and the company.
Who we are
We’re a stealth startup disrupting an entire industry, led by founders with proven expertise in product development and leadership - together with passionate engineers who have built state-of-the-art solutions for organizations of all sizes, with and without AI. Our hub is in central Stockholm, where we combine the energy of in-person collaboration with the flexibility of hybrid work.
The opportunity
We are looking for a driven AI/ML Engineer to join our team. This role blends development of GenAI powered features, evaluation, search algorithms and data ingestion pipelines. You’ll help shape how we integrate generative AI into real-world applications, delivering a value-creating AI first solution to end-users in short iterative feedback loops. You’ll be a key contributor, working closely with product, design, backend engineers to realize the future of our AI-powered platform.
You’ll employ Metrics Driven Development (MDD) and evaluation methodology to set up evaluation pipelines, helping us realize production-strength GenAI features covering both Agentic patterns and RAG. The focus is initially on utilizing GenAI solutions for the AI features, which will expand over time to incorporate ML solutions. You’ll have the opportunity to greatly influence both our GenAI and ML stack.
Key Responsibilities
- Design and build Gen AI features, covering both Agentic patterns and RAG. You’ll contribute with code to our backend platform.
- Design and build evaluation pipelines and metrics for public customer facing AI solutions.
- Design and build data ingestion pipelines from various sources (APIs, databases, files, customer systems)
- Design and build search algorithms, e.g., vector, semantic, graph, to support our AI features.
- Implement data quality monitoring and validation frameworks
- Integrate AI services and APIs into data processing workflows, including LLMs and vector embeddings
- Support both product data processing and internal business analytics as we scale
- Collaborate with other engineers on backend integration and system architecture
- Evaluate when GenAI or ML solutions are appropriate or whether simpler approaches suffice.
- Contribute to technical decisions about our evolving data stack and infrastructure
- Create and maintain data models optimized for both ML workloads and business analyticsBuild feature engineering pipelines to support ML model training and inference
Must-Have Skills
- Strong proficiency in Python
- Experience with Agentic AI, prompt engineering, tools.
- Understanding RAG, search algorithms, vector databases and embedding workflows.
- Experience with MDD and Gen AI evaluation using frameworks such as RAGAs or MLFlow.
- Experience with the full ML lifecycle.
- Experience with both structured and unstructured data
- Git proficiency and understanding of CI/CD concepts
Nice-to-Have Skills
- Experience as code contributor to backend components to a shipped app/service.
- Experience with NLP, deep learning and other types of ML models.
- Experience with MLOps.
- Experience with cloud platforms and their AI services
- Experience with graphs, and graph databases
- Experience with data monitoring and observability tools
- Knowledge of streaming data processing concepts
- Experience supporting business intelligence and analytics use cases
What we offer
From us you’ll get a rocketship in personal development - challenge and support to accelerate your ambitions even further.
And more:
- Early-team seat with real ownership in your domain.
- Fast feedback loops with the support and confidence to become the key contributor to complex decisions, both business and technical.
- Close collaboration with product, design, and engineering; low-ego, transparent ways of working.
- Own models from idea to production and see your work impact user experience and outcomes quickly.
- Department
- Product & Engineering
- Role
- AI/ML Engineer
- Locations
- Talent Venture Studios
- Remote status
- Hybrid
About Talent Venture Studios
About Talent Venture Studios
Talent Venture Studios is a global venture builder in HR tech and the future of work. We back early-stage teams with capital, networks, and hands-on expertise - helping them build products and companies that truly matter. Our mission is to transform how people and organizations grow, by building solutions that shape the future of work.