AI/ML Engineer
6.0/10
Improbable
Not specified
Remote
mid
10 days ago
aidevPythonFastAPIPydanticAIFastMCPMemgraphPostgresPyTorchUnslothAxolotlvLLM
AI Summary
The vacancy is well-defined but lacks compensation details, impacting overall attractiveness.
Check Match โ Just drop your CV
See your fit for AI/ML Engineer in seconds.
Description
Your Opportunity
- โข**Build production AI systems**: Design and implement the full stack, from FastAPI endpoints that handle requests, to training pipelines that process data, to inference services that serve predictions. You'll own the architecture, not just the model weights.
- โข**Train and deploy our DSLM**: Fine-tune models using Unsloth/Axolotl, but more importantly, build the robust infrastructure around it - data pipelines that feed training, evaluation frameworks that catch regressions, deployment systems that handle failover. Make it production-grade.
- โข**Integrate ML into our backend**: We use FastAPI, PydanticAI, FastMCP, Memgraph. You'll extend these systems with ML capabilities, not as a separate "ML service" but as a natural part of our backend architecture. Clean abstractions, proper error handling, observability.
- โข**Own inference performance**: Get models running fast, whether that's vLLM deployment, quantization strategies, batching optimizations, or caching. Hit our <200ms latency targets through engineering, not just throwing bigger GPUs at it.
- โข**Shape Project Genome's foundation**: Work with our Principal Engineer to architect how we ingest, process, and learn from global supply chain data. This is systems design as much as ML with data pipelines, graph databases, incremental learning strategies being just as important.
- โข**Mentor through code review and pairing**: Raise the bar on code quality, testing, and production practices across the team. Teach mid and junior engineers how to build ML systems that don't fall over.
Requirements
What we're looking for specifically
**Must have:**
- โข5+ years building production Python systems (backend services, APIs, data processing)
- โขStrong software engineering fundamentals: design patterns, testing, debugging, profiling
- โขExperience integrating LLMs into applications (OpenAI/Anthropic APIs, prompt engineering, streaming, PydanticAI)
- โขUnderstanding of ML training workflows (even if you're not an expert. You need to know enough to build the infrastructure)
- โขDocker, CI/CD, production deployment experience
- โขCan read and understand PyTorch code (you don't need to write novel architectures)
**Nice to have:**
- โขFine-tuning experience (LoRA, full fine-tuning, QLoRA)
- โขDistributed training basics (DeepSpeed, FSDP)
- โขGraph databases (Memgraph, Neo4j)
- โขSupply chain or logistics domain knowledge
- โขExperience with agent frameworks (LangChain, PydanticAI, etc.)
Loading similar jobs...