Member of Technical Staff - ML at Weekday AI

Bengaluru, karnataka, India -

Full Time

Start Date

Immediate

Expiry Date

14 Apr, 26

Salary

4000000.0

Posted On

14 Jan, 26

Experience

2 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

Skills

Python, LLMs, REST APIs, FastAPI, Flask, Machine Learning, Prompt Engineering

Industry

technology;Information and Internet

Description

This role is for one of the Weekday's clients Salary range: Rs 2500000 - Rs 4000000 (ie INR 25-40 LPA) Min Experience: 2 years Location: Bengaluru JobType: full-time We are looking for a driven Member of Technical Staff – Machine Learning with 2–3 years of hands-on experience in Python development, API engineering, and working with Large Language Models (LLMs). This role focuses on building and scaling ML-powered features, integrating AI capabilities into production systems, and contributing to the development of AI-first products. Key Responsibilities Design, build, and maintain Python-based machine learning pipelines and services. Develop, implement, and maintain RESTful APIs and integrations with internal and external systems. Integrate and operationalize LLM models (e.g., OpenAI, Anthropic, Gemini) within production applications. Create and optimize prompts, prompt templates, and prompt chains for various LLM-driven use cases. Collaborate closely with product and engineering teams to translate business requirements into ML solutions. Monitor model performance and optimize inference, latency, and reliability. Follow best practices for code quality, testing, scalability, and maintainability. Contribute to system design and architecture discussions for AI-driven platforms and products. Required Skills & Experience Strong proficiency in Python, with experience using FastAPI or Flask. Solid experience in API development and integration, including REST standards, authentication, versioning, and documentation. Hands-on experience with LLM integration, embeddings, and vector-based retrieval workflows. Good understanding of prompt engineering, including system prompts, RAG workflows, and evaluation techniques. Experience handling JSON schemas, logging, asynchronous programming, and error handling. Foundational knowledge of machine learning concepts such as preprocessing, evaluation, and deployment. Exposure to cloud platforms (AWS, GCP, or Azure) is an advantage. Nice-to-Have Experience with vector databases such as Pinecone, Weaviate, or FAISS. Familiarity with frameworks like LangChain, LlamaIndex, or similar tools. Understanding of CI/CD pipelines and modern DevOps practices. Exposure to monitoring, observability, and model performance tracking tools. Basic knowledge of Docker and containerized deployments. Qualifications Bachelor’s degree in Computer Science, Engineering, or a related field. 2–3 years of professional experience in software or machine learning engineering roles, preferably in a product-based environment. Skills Python, LLMs, REST APIs, FastAPI, Flask, Machine Learning, Prompt Engineering

Responsibilities

The role involves designing, building, and maintaining Python-based machine learning pipelines and services, as well as developing and implementing RESTful APIs. Additionally, the candidate will integrate and operationalize LLM models within production applications.