Artificial Intelligence Engineer

Yotta Data Services Private Limited; Mumbai; On-site; INR 3,500,000 - 7,000,000; Full time

Job summary

Yotta Data Services Private Limited in Mumbai seeks a senior AI/ML Lead to spearhead research, design, and deployment of next-generation AI features for a no-code platform and scalable orchestration system. You will build scalable ML pipelines, optimize production models, mentor team members, and translate research into product features aligned with business goals.

Collaboration with product, UX, and engineering teams is essential.

Qualifications

• Advanced degree or equivalent in CS/EE/Math/Statistics with AI specialization.
• Strong publication record in AI/ML conferences preferred.
• Experience building and deploying deep learning models including transformers.

Responsibilities

• Design and implement ML/DL models for NLP, CV, and generative AI for no-code platforms.
• Develop and optimize large-scale models and transformer architectures.
• Architect end-to-end ML pipelines: data processing, training, deployment, monitoring.
• Collaborate with software teams to productionize models for reliability and scalability.
• Lead AI research efforts contributing to IP, patents, and publications.
• Provide mentorship to junior AI/ML staff and guide cross-functional teams.

Skills

No-code AI platforms

LangChain

Probabilistic modeling

MLOps practices

Cloud platforms

Open-source contributions

AI ethics & privacy

Academic publications

Education

Bachelor's or higher in CS/EE/Math/Statistics with AI specialization

Tools

Python

PyTorch

TensorFlow

JAX

Docker

Kubernetes

LangChain

CI/CD tooling

Job description

Lead the research, design, and development of advanced AI and ML models powering a cutting edge AI-driven no-code development platform and a scalable AI inference and training orchestration system. Responsible for building scalable ML pipelines, optimizing models for production, mentoring team members, and translating research innovations into impactful product features aligned with business goals.

Job Responsibilities:

• Design and implement state-of-the-art machine learning and deep learning models for NLP, computer vision, and generative AI relevant to no-code AI coding and AI orchestration platforms.
• Develop, optimize, and fine-tune large-scale models, including transformer-based architectures and generative models.
• Architect and manage end-to-end machine learning pipelines: data processing, training, evaluation, deployment, and continuous monitoring.
• Collaborate closely with software engineering teams to productionize models ensuring reliability, scalability, and performance.
• Research and integrate cutting-edge AI techniques and algorithms to maintain product competitiveness.
• Lead AI research efforts contributing to intellectual property generation, patents, and academic publications.
• Provide technical leadership and mentorship to junior AI/ML team members.
• Collaborate cross-functionally with product managers, UX designers, and engineers to deliver AI-powered product features.
• Maintain up-to-date knowledge of AI research trends and technologies, assessing their applicability.
• Ensure compliance with data privacy and security standards in AI model development.

Good to have skills:

• Experience with AI-driven no-code platforms or automated code generation.
• Familiarity with AI workflow orchestration frameworks like LangChain, Crew AI, or similar.
• Knowledge of probabilistic modeling and uncertainty quantification.
• Hands-on experience with MLOPs tools and practices, including CI/CD, model versioning, and monitoring.
• Familiarity with cloud platforms (AWS, GCP, Azure) and container orchestration (Docker, Kubernetes).
• Contributions to open-source AI projects or patent filings.
• Understanding of AI ethics, data privacy (GDPR, SOC2) compliance.
• Strong academic research background with publications in top-tier AI/ML conferences.

Qualification and Experience:

• Equivalent degree in Computer Science, Electrical Engineering, Statistics, Mathematics, or related fields with a specialization in Artificial Intelligence, Machine Learning, or Deep Learning.
• Strong research publication record in reputed AI/ML conferences (NeurIPS, ICML, ICLR, CVPR, ACL).
• Demonstrated experience in developing and deploying deep learning models including transformers, CNNs, RNNs, GNNs, and generative AI models.
• Proven skills in NLP and/or computer vision.
• Hands-on experience with Python and ML frameworks such as PyTorch, TensorFlow, JAX.
• Experience building scalable ML pipelines and applying MLOPs best practices.
• Knowledge of distributed training, GPU acceleration, and cloud infrastructure is highly desirable.
• Excellent problem-solving, analytical, and communication skills.
• Experience mentoring or leading junior AI researchers/engineers is a plus.
• Prior exposure to AI-driven no-code platforms, AI orchestration frameworks, or automated code generation technologies is beneficial.