AETRIS-AI Labs

AI/ML Engineer

Remote Internship AETRIS-AI Labs

Design, train, and fine-tune custom machine learning models, handle LLM orchestration (RAG, prompt engineering), and help hold model accuracy to a production bar.

Responsibilities

  • Design, train, and fine-tune machine learning models for real product features
  • Build and iterate on LLM orchestration — retrieval-augmented generation (RAG), prompt engineering, and evaluation pipelines
  • Benchmark model accuracy against a production bar, not a demo bar
  • Collaborate with product and engineering to turn model outputs into reliable user-facing features

Requirements

  • Solid foundation in Python and core ML concepts
  • Familiarity with at least one ML framework (PyTorch, TensorFlow, or similar)
  • Comfortable reading research papers and translating ideas into working code
  • Currently pursuing or recently completed a degree in CS, ML, or a related field

Skills

PythonPyTorch / TensorFlowLLM APIsRAG pipelinesPrompt engineering

Benefits

  • Mentorship from senior AI engineers
  • Real shipped features on AxisFolio
  • Certificate on completion
  • Flexible, remote-first schedule

AETRIS-AI Labs designs, builds, and ships AI-native products — starting with AxisFolio, our portfolio intelligence engine. We’re a small, remote-first team that ships real products rather than running simulated projects.

Turn AI research into products people actually use every day, engineered to a production bar from day one.

Keep exploring.