Machine Learning Engineer
Build and ship ML and GenAI features — train and evaluate models, build RAG and agent pipelines, and expose them through production APIs. 1-3 years experience.
About the role
You will build the models and pipelines behind WebbyButter's AI products: classical ML where it fits, GenAI where it wins. Expect a mix of training and evaluating models, building RAG and agent pipelines, and exposing everything through reliable APIs.
What you will do
- Train, evaluate and iterate on ML models for prediction, classification and document understanding
- Build RAG and agent pipelines using LLM APIs and open-source models
- Design datasets, labelling workflows and evaluation metrics
- Serve models and pipelines through FastAPI services with proper logging and tests
- Collaborate with full-stack engineers to integrate ML features into client applications
What we are looking for
- 1-3 years of hands-on experience with Python for ML or backend work
- Working knowledge of PyTorch or TensorFlow and scikit-learn
- Experience with at least one LLM API and with embeddings or vector search
- SQL and data wrangling with Pandas
- Comfort working in a fast-paced, collaborative team
Nice to have
- MLOps tooling (MLflow, Weights & Biases, Docker, cloud ML services)
- Vector databases such as pgvector, Qdrant or Pinecone
- Published projects, Kaggle work or open-source contributions
Key skills
About WebbyButter AI
WebbyButter is an AI-first software company headquartered in Vadodara, Gujarat, India. We build production AI systems for enterprises and funded startups: RAG pipelines, AI agents, intelligent document processing, computer vision and the web products around them. All current roles are based at our Vadodara office.