Senior ML Research Engineer

The Role

 

We’re looking for a Senior ML Research Engineer to lead the design, implementation, and optimization of large-scale models powering intelligent agents. Your role spans model architecture, shipping, and production monitoring.

 

Key Responsibilities

 

  • Conduct cutting-edge research on agentic AI systems — exploring new methods in reasoning, memory, planning, and multi-agent collaboration — and translate findings into production-ready prototypes.
  • Design, fine-tune, and evaluate large language models and neural networks for modular agent behavior
  • Build robust ML pipelines (data ingestion, feature engineering, model training, serving, monitoring) 
  • Implement evaluation frameworks, A/B testing, and performance metrics to quantify agent effectiveness
  • Ensure reproducibility, traceability, and compliance across ML lifecycle
  • Collaborate with backend engineers to define inference service SLAs and efficient real-time ML delivery
  • Mentor junior ML teammates and establish team-level best practices

 

 

Requirements

 

  • 5+ years as an ML Engineer or applied ML researcher with production model deployment experience
  • Strong Python skills, experience with ML frameworks (TensorFlow, PyTorch) and LLM tooling (HuggingFace)
  • Familiarity with data pipelines, SQL, cloud platforms (AWS)
  • Solid understanding of model evaluation, A/B testing, and ML performance metrics
  • Excellent collaboration skills with product, backend, and data teams

 

Nice to Haves

 

  • Prior work with conversational agents, retrieval-augmented generation, or multi-model orchestration
  • Experience with vector search stacks (e.g. Pinecone, FAISS)
  • Reinforcement Learning experience
  • Startup experience and ability to navigate ambiguity and shape technical direction

 

Why You Should Join

  • Competitive equity & pay - get in early and own what you build.
  • Work closely with experienced founders with a proven startup track record.
  • Move fast, ship fast - no corporate bureaucracy.
  • Shape the AI revolution in healthcare - massive market, untapped potential

Required languages

English B2 - Upper Intermediate
Python, Machine Learning, PyTorch, Deep Learning, AI Agents, Multi-agent LLMs, Multi-agent system, LLM, Monitoring, Retrieval-Augmented Generation
Published 24 June · Updated 11 November
Statistics:
34 views
·
2 applications
To apply for this and other jobs on Djinni login or signup.
Loading...