About The Job
As an AI/ML Engineer, you will work on building Kuku's next-generation AI-powered content creation platform, with a focus on LLMs, multimodal AI, content understanding, and AI-generated microdramas. You will work with large-scale microdrama transcripts and build ML systems that can understand stories, characters, scenes, and narrative structures, and use this knowledge to generate and optimize new content.
This role provides an prospect to work across Generative AI, LLMs, instruction-tuning data, agentic workflows, screenplay generation, and AI video generation, while building production-grade ML systems used by Kuku's content ecosystem.
Responsibilities
- LLM &
- GenAI: Build LLM-powered systems for screenplay generation, dialogue generation, story continuation, summarization, rewriting, critique, and content transformation.
- Instruction-Tuning Data: Build and curate high-quality instruction-tuning and evaluation datasets from Kuku's large-scale microdrama transcript corpus.
- Content Understanding: Develop ML/LLM pipelines to extract characters, relationships, scenes, story arcs, dramatic beats, emotions, conflicts, twists, and other structured information from content.
- AI Screenwriting: Build systems that transform story ideas and structured story information into episodes, scenes, dialogue, and production-ready screenplays.
- AI Video Generation: Develop workflows to convert screenplay and scene information into structured prompts and instructions for AI video generation models such as Seedance.
- Agentic Systems: Build multi-step LLM workflows using frameworks such as LangGraph or LangChain for story planning, screenplay generation, quality checking,
continuity validation, and video prompt generation.
- Embeddings &
- Retrieval: Build semantic retrieval systems for characters, stories, scenes, dialogue, and other content using embedding models and vector databases.
- Model Experimentation: Experiment with different LLMs, VLMs, embedding models, fine-tuning techniques, prompting strategies, and retrieval approaches to improve quality, latency, and cost.
- Evaluation: Develop automated and human-in-the-loop evaluation pipelines to measure story quality, narrative consistency, character consistency, and AI-generated content quality.
- ML Execution: Build and productionise ML systems end-to-end, including data pipelines, model inference, experimentation, monitoring, and optimization.
- Collaboration: Work closely with senior ML engineers, content teams, product managers, and backend engineers to translate creative and product problems into scalable ML solutions.
Preferred Qualifications
- Education & Experience: Bachelor's or Master's in Computer Science, Machine Learning, Statistics, or a related engineering field, with 2-4 years of relevant experience and a proven ability to build and deploy ML systems
- LLM &
- GenAI: Hands-on experience with LLM APIs, open-source LLMs, prompt engineering, RAG, embeddings,
or generative AI applications.
- Model Training: Experience training or fine-tuning ML/deep learning models using frameworks such as PyTorch or TensorFlow. Familiarity with Hugging Face Transformers and LoRA/PEFT is a plus.
- Production Experience: Experience building and deploying production ML models or GenAI applications, including data pipelines, APIs, monitoring, and cloud infrastructure.
- NLP: Strong understanding of transformers, embeddings, semantic search, text classification, sequence modeling, or related NLP techniques.
- Agentic Systems: Experience with LLM orchestration frameworks such as LangGraph, LangChain, or equivalent technologies is desirable.
- Multimodal AI: Experience with VLMs, speech/audio models, image generation, or video generation is a strong plus.
- ML Fundamentals: Strong understanding of machine learning, deep learning, experimentation, and model evaluation.
- Engineering: Strong Python programming and software engineering skills with the ability to build scalable and maintainable ML systems.
- Data &
- Infrastructure: Experience with SQL, data processing, vector databases, cloud platforms, Docker, Kubernetes, or similar technologies is desirable.
- Research Awareness: Stay up-to-date with advancements in LLMs, multimodal AI, generative AI, and applied machine learning.
Skills: docker,python,kubernetes,prompt engineering,generative ai,deep learning,embedding,vector database,langgraph,statistics,pytorch,langchain,ml evaluation,ml system,ai agent system,ai multimodal,tensorflow,rag,llm,cloud infrastructure,machine learning,natural language processing
📌 AI/ML Engineer (Mumbai)
🏢 Kuku
📍 Mumbai