Responsibilities:
Develop and optimize adaptive learning models, recommendation engines, and AI reasoning modules for education and workforce products.
Build offline-capable AI systems that work in low-connectivity environments using edge models and lightweight inference.
Train NLP and speech models for multilingual support (Hindi, Marathi, Gujarati, English, etc).
Research, design, and deploy algorithms for profiling, assessment, and personalized learning paths.
Collaborate with product, design, and engineering teams to integrate AI features into apps and simulations.
Conduct experiment design, model evaluation, AB testing, and continuous model improvement.
Maintain clean documentation, model pipelines, datasets, and training processes.
Ensure ethical AI development with strong focus on safety, accuracy, bias reduction, and data privacy.
Stay updated with the latest advancements in GenAI, LLMs, and on-device AI.
Qualifications:
Bachelor’s or Master’s in Computer Science, AI/ML, Data Science, or related fields.
Strong hands-on experience with Python, TensorFlow, PyTorch, Scikit-learn, or similar ML frameworks.
Experience in building NLP, speech processing, or recommendation models.
Familiarity with LLMs, model fine-tuning, quantization, and optimization for edge devices.
Understanding of data pipelines, model training workflows, and evaluation methodologies.
Bonus points for experience with:
Offline/on-device AI (e.g., TensorFlow Lite, ONNX, GGUF models)
Educational technology systems
Bilingual/multilingual models
Real-time inference optimization
Strong problem-solving, experimentation mindset, and ability to work in fast-paced environments.
Compensation & Benefits:
Competitive industry-aligned salary package
Performance-based bonuses and stock options (role-dependent)
Opportunity to work on impactful AI products used across India
Flexible work culture (Hybrid)
Annual learning & upskilling budget for certifications, courses & conferences
Rapid career growth in a product-driven, innovation-first environment
Health benefits and wellness support
Work directly with founders, research teams, and domain experts