Magnit

Who We Are

Magnit is the future of work. Serving hundreds of the world’s most recognizable brands for the past 30+ years, Magnit offers the industry’s first holistic platform for the modern workforce. Magnit's integrated workforce management (IWM) platform supported by data, software, intelligence, and best-in-class services team is key to our clients’ success. It can adapt quickly to regional or industry economic shifts, and provides the speed, scale, flexibility, transparency, and expertise required to meet an organization’s contingent workforce management, talent strategy and broader organization goals. At Magnit, you’ll work with passionate colleagues who collaborate and deliver meaningful results that positively transform the largest companies around the globe.

Senior AI & ML Platform Engineer

Location: Bengaluru

Experience Level: 3–10 Years

About the Role

We are building next-generation AI-powered platforms that leverage Large Language Models (LLMs), advanced Machine Learning (ML), NLP, and agentic workflows to automate critical business processes across multiple domains.

We are looking for experienced AI/ML Engineers who are passionate about building real-world AI products—not just experimenting. You will help design, prototype, and deploy intelligent systems integrating LLMs, agents, APIs, and robust ML models for classification, prediction, and optimization.

Responsibilities

• Build and integrate agentic AI workflows using LangChain or similar frameworks.
• Develop APIs and backend services to support LLM and ML-driven platforms.
• Implement, train, and optimize lightweight and scalable ML models for classification, prediction, and anomaly detection.
• Work with LLMs (OpenAI, Claude, Gemini) to fine-tune prompts, create retrieval-augmented generation (RAG) systems, and chain multi-step tasks.
• Design and optimize vector search implementations for LLM-driven applications.
• Integrate AI models with external systems (VMS platforms, ERP systems) using APIs.
• Implement document processing pipelines combining OCR with LLM/ML validation.
• Apply ML techniques for supply-demand forecasting, fraud detection, and optimization.
• Participate in design discussions, brainstorming sessions, and agile sprints.
• Collaborate with senior AI architects and SMEs to refine platform capabilities.

Required Skills

• Strong experience in Python for ML and AI development.
• Expertise in ML frameworks such as TensorFlow, PyTorch, scikit-learn.
• Knowledge of NLP tasks (text extraction, classification, summarization).
• Experience building and optimizing RAG workflows with vector databases.
• Solid understanding of API development (FastAPI, Flask, or similar frameworks).
• Familiarity with prompt engineering and agent orchestration concepts.
• Ability to build scalable ML models for classification, forecasting, and anomaly detection.
• Experience integrating ML solutions with cloud services (AWS, Azure, GCP).
• Knowledge of OCR tools and techniques for document processing.
• Exposure to multi-agent architectures and reinforcement learning is a plus.
• Understanding of feature engineering, model evaluation, and optimization techniques.

Mindset We Are Looking For

• Builder’s mindset: You love turning ideas into working solutions fast.
• Curiosity: Passionate about keeping up with the latest in GenAI and ML.
• Problem-solver: Comfortable working in ambiguous environments where solutions need discovery, not just execution.
• Collaboration: Open to learning from senior architects and iterating based on feedback.

What Magnit will Offer You

At Magnit, you’ll be joining an innovative, high-growth environment and can quickly make an impact to help transform the largest companies in the world. You will work with passionate colleagues who collaborate and deliver. Magnit offers all employees the opportunity for growth and development, and we want individuals to fulfill their potential and blaze their own trails!

Magnit will offer you a competitive PTO and benefits package, including medical, dental, and vision coverage, retirement planning, as well as discounts and perks for tickets, travel, merchandise and more! Magnit encourages employees to participate in giving back, and we will match employee contributions to favorite charities and support corporate volunteering hours to make a difference in your community!

If this role isn’t for you

Stay in touch, we will let you know when we have new positions on the team.

To see a complete list of our open career opportunities please visit.

https://magnitglobal.com/us/en/company/careers.html

To do our best work we need different viewpoints. Therefore, we celebrate diversity and embrace inclusion.

As an equal opportunity employer, we are dedicated to building a team that represents a variety of backgrounds, perspectives, and skills. We strive to ensure that we maintain a positive and enriching work environment for all.

By applying to this role, you consent to Magnit safely storing and managing your personal data. Please read this link to learn more.

https://magnitglobal.com/us/en/privacy-notice.html