Artificial Intelligence and Machine Learning Engineer

Taskus India Private Limited; Mumbai; On-site; INR 1,500,000 - 2,500,000; Full time

Job summary

TaskUs in Mumbai is seeking an AI/ML Engineer with 1 to 4 years of experience to design and implement AI-powered solutions, leveraging NLP, NLU, LLM, and Generative AI technologies. The role involves building scalable models and collaborating with product and data teams to deliver robust AI-based products.

You will work with cutting-edge AI tech, mentor junior engineers, and help drive high-quality outcomes in a fast-paced, People First environment. Enthusiasm for innovation is essential.

Qualifications

• Bachelor's or Master's degree in Computer Science, Information Technology or related fields.
• 1 to 4 years of experience in machine learning and artificial intelligence.
• Hands-on experience in building and deploying AI-based products using NLP, NLU, LLM, and Generative AI technologies.
• Strong hands-on experience in Object-Oriented Programming System, Python, FastAPI, Generative AI Prompting. Experience in containerization (Docker), Jenkins.
• Experience in leading and managing a team of AI and Machine Learning Engineers.
• Experience in AWS AI Services would be added advantage.
• Excellent problem-solving skills with the ability to think creatively and critically.
• Strong communication and presentation skills with the ability to communicate technical solutions to non-technical stakeholders.
• Experience in Agile development methodologies and project management.

Responsibilities

• Design and develop machine learning models to solve complex business problems using AI, NLP, NLU, LLM, and Generative AI technologies.
• Develop and implement algorithms for data processing, feature extraction, and model training.
• Collaborate with cross-functional teams to identify business requirements and create solutions that meet those requirements.
• Identify and evaluate new AI technologies and tools that can improve the performance and efficiency of our products.
• Manage and mentor a team of AI and Machine Learning Engineers.
• Ensure that the products are delivered on time, meet quality standards, and are scalable.
• Communicate technical solutions and results to non-technical stakeholders.

Skills

LangChain

Multi-agent systems

Embeddings

Vector databases

LLMs / GenAI

Python

FastAPI

Docker

AWS

Jenkins

Education

Bachelor's or Master's in CS/IT

Tools

Docker

Jenkins

GitHub Actions

FastAPI

AWS (SageMaker)

Job description

AI/ML Engineer

About TaskUs: TaskUs is a provider of outsourced digital services and next-generation customer experience to fast-growing technology companies, helping its clients represent, protect and grow their brands. Leveraging a cloud-based infrastructure, TaskUs serves clients in the fastest-growing sectors, including social media, e-commerce, gaming, streaming media, food delivery, ride-sharing, HiTech, FinTech, and HealthTech. The People First culture at TaskUs has enabled the company to expand its workforce to approximately 60,000 employees globally. Presently, we have a presence in twenty-three locations across twelve countries, which include the Philippines, India, and the United States. It started with one ridiculously good idea to create a different breed of Business Processing Outsourcing (BPO)! We at TaskUs understand that achieving growth for our partners requires a culture of constant motion, exploring new technologies, being ready to handle any challenge at a moment’s notice, and mastering consistency in an ever-changing world.

What We Offer

At TaskUs, we prioritize our employees' well-being by offering competitive industry salaries and comprehensive benefits packages. Our commitment to a People First culture is reflected in the various departments we have established, including Total Rewards, Wellness, HR, and Diversity. We take pride in our inclusive environment and positive impact on the community. Moreover, we actively encourage internal mobility and professional growth at all stages of an employee's career within TaskUs. Join our team today and experience firsthand our dedication to supporting People First.

What can you expect in a Developer, AI/ML Engineer role with TaskUs

We are looking for an Artificial Intelligence and Machine Learning Engineer with 1 to 4 years of experience with python technology, who will be responsible for leveraging AI, NLP, NLU, LLM, and Generative AI technologies to create innovative products. The ideal candidate should have a strong background in machine learning and artificial intelligence with hands‑on experience in building scalable and reliable AI‑based products.

Key Responsibilities

• Design and develop machine learning models to solve complex business problems using AI,NLP, NLU, LLM, and Generative AI technologies
• Develop and implement algorithms for data processing, feature extraction, and model training
• Collaborate with cross-functional teams to identify business requirements and create solutions that meet those requirements
• Identify and evaluate new AI technologies and tools that can improve the performance and efficiency of our products
• Manage and mentor a team of AI and Machine Learning Engineers
• Ensure that the products are delivered on time, meet quality standards, and are scalable
• Communicate technical solutions and results to non‑technical stakeholders

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Information Technology or related fields
• 1 to 4 years of experience in machine learning and artificial intelligence
• Hands‑on experience in building and deploying AI‑based products using NLP, NLU, LLM, and Generative AI technologies
• Strong hands‑on experience in Object‑Oriented Programming System, Python , FastAPI , Generative AI Prompting . Experience in containerization (Docker), Jenkins.
• Experience in leading and managing a team of AI and Machine Learning Engineers
• Experience in AWS AI Services would be added advantage
• Excellent problem‑solving skills with the ability to think creatively and critically
• Strong communication and presentation skills with the ability to communicate technical solutions to non‑technical stakeholders
• Experience in Agile development methodologies and project management

Some key skills

• Agentic Frameworks - LangChain, Multi‑agent systems , Episodic memory, Semantic memory , tool orchestration.
• Vector Databases / RAG - Vector DB (Anyone) , Embeddings , Hybrid search (semantic + keyword) , Chunking, context window optimization ,Metadata filtering, Hybrid pipelines
• LLMs / GenAI - Prompt engineering , Fine‑tuning, LoRA, PEFT , Guardrails , Context window.
• Infrastructure - AWS (S3, Lambda, ECS, EKS, SageMaker) , Docker , CI/CD pipelines (Jenkins, GitHub Actions) , Model serving (FastAPI), Secrets management (AWS Secrets Manager) , GPU acceleration (CUDA)

Education / Certifications

• Bachelor's/College Degree in Engineering Computer Science/Information Technology, or equivalent.

DEI

In TaskUs we believe that innovation and higher performance are brought by people from all walks of life. We welcome applicants of different backgrounds, demographics, and circumstances. Inclusive and equitable practices are our responsibility as a business. TaskUs is committed to providing equal access to opportunities. If you need reasonable accommodations in any part of the hiring process, please let us know.