Artificial Intelligence Engineer

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

Job summary

Yotta Data Services Private Limited seeks an AI/ML leader to drive the research, design, and deployment of advanced models powering a no-code AI platform. You will architect scalable ML pipelines, optimize models for production, and mentor junior researchers.

You will oversee research efforts, publish in top AI venues, and collaborate with product, UX, and engineering teams to deliver impactful features while upholding privacy and security standards.

Qualifications

• Equivalent degree in CS/EE/Statistics/Math with AI/ML/DL specialization.
• Strong publication record in AI/ML conferences (NeurIPS/ICML/ICLR/CVPR/ACL).
• Experience building and deploying DL models (transformers, CNNs, RNNs, GNNs, generative AI).
• Proficiency in NLP and/or computer vision; Python and ML frameworks (PyTorch, TensorFlow, JAX).
• Experience with scalable ML pipelines and MLOps practices; distributed training; cloud infra.

Responsibilities

• Lead design and development of ML models for NLP, CV, generative AI in a no-code AI platform.
• Architect end-to-end ML pipelines: data processing, training, deployment, monitoring.
• Collaborate with SWE to productionize models for reliability and performance.
• Mentor junior AI/ML team members and guide research direction.
• Drive AI tech innovations, contribute to IP, patents, and publications.
• Stay updated on AI trends and ensure data privacy and security in development.

Education

Equivalent degree in Computer Science, Electrical Engineering, Statistics, Mathematics, or related fields with specialization in AI/ML/DL

Tools

LangChain

Crew AI

Docker

Kubernetes

CI/CD

PyTorch

TensorFlow

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.