AI/ML Engineer

AI / ML Engineer (LLM & RAG)

Location: In-Office

Experience: 2–5 years

Role Overview - Build and optimize the AI brain of Astro AI using LLMs, RAG pipelines, and structured astrology datasets.

Responsibilities

• Develop LLM-based astrology interpretation models
• Build Retrieval-Augmented Generation (RAG) systems
• Optimize prompt engineering and response accuracy
• Integrate APIs (OpenAI / open-source LLMs)
• Improve latency, cost efficiency, and hallucination control

Requirements

• Strong Python background
• Experience with LLMs, embeddings, vector DBs (FAISS / Pinecone)
• Understanding of NLP pipelines
• Bonus: prior AI product experience