SDE – Machine Learning (AI Agents)
Location : Remote | Type : Paid | Experience : 0 to 3 years (Hands-on exp with AI Agents, LLMs, or RAG)
What We're Looking For :
We need an exceptionally strong coder with real-world experience in LLMs, RAG systems, and multi-agent orchestration. You must be curious, fast-learning, and excited to build with a high-performance startup team.
Who We Are :
Steps AI builds advanced GenAI products that power agentic data retrieval and automation for enterprises. We work on LLMs, multi-agent systems, RAG, LLMOps, and generative AI orchestration.
Key Responsibilities :
Develop LLM-based agents for enterprise use cases
Build scalable RAG pipelines (Naive, Graph, Struct)
Implement LLMOps and production-ready ML infrastructure
Optimize LLM reasoning with CoT, ToT, GoT prompting
Design and deploy multi-agent orchestration with tools like LangGraph, LangChain, or AutoGen
What You Must Know :
Solid grasp of LLMs (GPT, LLaMA, etc.) and transformer architectures
Hands-on experience with RAG systems and agentic workflows
Familiarity with Docker, MLOps, cloud deployment (AWS / Azure)
Strong coding skills, system design, and software architecture principles
Proficiency in Python, PyTorch / TensorFlow, Hugging Face, LangChain, etc.
Ability to work collaboratively and own outcomes
Why Join Us :
Work on cutting-edge AI products, not toy projects
Build with serious engineers in a fast-moving startup
Mentorship, autonomy, and future full-time opportunity
⚠️ Important : Only apply if you have hands-on experience with LLMs, AI Agents, or RAG. No exceptions.
Steps AI – We co-create the AI-native enterprise.
Artificial Intelligence Engineer • Kollam, Kerala, India