HeyMilo

Technology

AIEngineer

$3000–5500k ~AI est. Colombo, Sri Lanka FULL TIME Remote Friendly
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Mid+ candidates.

The Brief

“AI Engineer at HeyMilo. Skills: AI systems, LLMs, Prompt engineering, AI pipelines. Write Python code. Design FastAPI-based APIs”

Industry & Context.

Technology
Problems you'll solve

Problem-solving mindset

What They're Looking For.

Must Have

Hands-on experience in Python, Hands-on experience in FastAPI, Deep understanding of LLMs, Deep understanding of prompt engineering, Deep understanding of AI infrastructure, Proven experience with Agentic AI frameworks, Proven experience with data pipelines, Proven experience with asynchronous systems, Solid computer science fundamentals, Problem-solving mindset

Nice to Have

Fast learner, Curious, Adaptable, Exploring AI innovation frontier

What You'll Do.

Design FastAPI-based APIs

Implement FastAPI-based APIs

Maintain FastAPI-based APIs

Transform ideas into features

Apply prompt engineering

Minimize hallucinations

Improve LLM performance

Build contextual search

Build AI workflow pipelines

Optimize contextual search

Optimize AI workflow pipelines

Fine-tune deep learning models

Deploy deep learning models

Optimize deep learning models

Research emerging AI tools

Research emerging AI frameworks

Develop observability systems

Develop AI content auditing systems

How You'll Work.

Team & Collaboration

World-class engineers

Full Job Description

At HeyMilo, we're redefining how companies recruit using Agentic AI to automate interviews, evaluate candidates, and scale intelligent hiring experiences.   We're looking for an AI Engineer who's passionate about building next-generation AI systems that create real-world impact. If you love experimenting with LLMs, deploying AI pipelines, and pushing the boundaries of whats possible youll fit right in.   What You'll Do - Write clean, scalable, and maintainable Python code that powers HeyMilo's AI-driven platform. - Design, implement, and maintain FastAPI-based APIs to deliver intelligent, enterprise-grade AI services. - Prototype and iterate rapidly, transforming new ideas into production-ready features. - Experiment with LLMs and apply robust prompt engineering to minimize hallucinations and improve performance. - Build and optimize Retrieval-Augmented Generation (RAG), contextual search, and AI workflow pipelines. - Fine-tune, deploy, and optimize deep learning models for NLP and computer vision applications. - Research emerging AI tools, frameworks, and research adopting them to enhance HeyMilo as a product. - Develop observability and AI content auditing systems to ensure reliability, transparency, and continuous improvement. Who You Are - Strong in Python, with hands-on experience in FastAPI. - Deep understanding of LLMs, prompt engineering, and AI infrastructure. - Proven experience with Agentic AI frameworks, data pipelines, and asynchronous systems. - A fast learner curious, adaptable, and always exploring the frontier of AI innovation. - Solid computer science fundamentals and a problem-solving mindset that thrives in a fast-paced, startup environment. Why Join HeyMilo - Work on cutting-edge AI technologies shaping the future of recruiting. - Collaborate with world-class engineers from Instagram, Salesforce, Uber, and Meta. - High ownership, high impact build, ship, and see your work in action fast. - Competitive compensation, strong performance-based rewar

Free ATS check

Applying for this AI Engineer role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

How to Apply on Ashby

  • Ashby is a fast modern ATS — most applications take under 3 minutes.
  • The resume parser is strong; verify parsed experience dates and job titles.
  • Custom screening questions are often scored algorithmically — answer completely.
  • Location field affects geo-based screening; use your actual metro area.

ANONYMOUS · UNFILTERED

What do employees actually say about HeyMilo?

Real rants from real employees. Read before you apply.

Read Company Rants →