Improbable

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...