AI / ML Engineer
Kainjoo Sa
Morges, Switzerland
2 days ago
Role details
Contract type
Permanent contract Employment type
Full-time (> 32 hours) Working hours
Regular working hours Languages
English, French Experience level
Intermediate Compensation
CHF 208KJob location
Remote
Morges, Switzerland
Tech stack
API
Artificial Intelligence
Google BigQuery
Data Security
Document Management Systems
Python
PostgreSQL
Machine Learning
TensorFlow
Azure
Service Development Studio
Feature Engineering
Flask
Large Language Models
Prompt Engineering
Backend
FastAPI
Build Management
AI Platforms
Scikit Learn
Information Technology
XGBoost
Machine Learning Operations
Virtual Agents
REST
Job description
You'll build Python-based AI services, implement LLM-powered features, and work with backend teams to expose AI outputs through clean APIs. You'll also help define what's possible - advising on feasibility, trade-offs, and the right tools for each use case.
- Design and build Python-based AI/ML services (FastAPI or Flask) for intelligent product features
- Implement LLM-powered features: classification, extraction, summarisation, scoring, and agent-based workflows
- Train, validate, and iterate on ML models using structured and time-series data where applicable
- Build and maintain RAG pipelines for knowledge-intensive use cases
- Expose AI outputs via REST APIs consumable by application backends
- Implement feature engineering pipelines from relational data sources
- Monitor model and AI system performance in production - detect drift, evaluate outputs, retrain as needed
- Collaborate with product and engineering teams to define AI-driven feature requirements
- Document system logic and data requirements clearly for non-ML stakeholders
Requirements
Personality
- You're excited about applied AI - not just the research, but shipping things that actually work
- You're rigorous: you evaluate outputs, you catch hallucinations, you don't ship blindly
- You communicate well about what AI can and can't do, especially with non-technical stakeholders
- You stay close to the field - you know what's changed in the last six months
- You're collaborative and enjoy working at the intersection of product and engineering
Experience
- 3+ years of AI/ML engineering or applied AI experience
- Strong Python skills - FastAPI or Flask for service development
- Hands-on experience integrating LLM APIs (OpenAI, Anthropic, or similar) into production systems
- Experience building or working with RAG architectures and vector databases (Pinecone, Weaviate, pgvector, or similar)
- Familiarity with ML frameworks for structured data tasks (scikit-learn, XGBoost, or similar)
- Experience deploying AI/ML services as REST APIs in production
- Ability to evaluate and monitor AI output quality in production
Nice to Have
- Experience with prompt engineering, fine-tuning, or instruction tuning
- Background in operations research or constraint-based optimisation
- Familiarity with PostgreSQL or BigQuery for data access
- Knowledge of agentic AI patterns (tool use, multi-step reasoning, function calling)
- Experience with model monitoring tools (MLflow, Weights & Biases), * Bachelor's or Master's degree in Computer Science, Mathematics, Data Science, or a related field
- Relevant AI/ML certifications or published work are a plus
- Fluent in English; French is a plus
About the company
AI is becoming a core part of what we build at Kainjoo - not as a buzzword, but as real, useful features embedded in the products our clients depend on. You'll be the person who makes that happen: designing and integrating AI and machine learning capabilities into production systems, working closely with backend teams, and pushing the quality bar for what intelligent features can do.