Job
Description
As an AI Engineer at our company in Kochi, you will play a vital role in designing, prototyping, and delivering generative AI capabilities. Your responsibilities will include working on practical research, building POCs, fine-tuning open models, contributing to multimodal experiments, and taking solutions towards production. Here is an overview of what you will do: - Build and evaluate prototypes / POCs for generative AI features and ideas. - Fine-tune and adapt open-source LLMs and smaller generative models for targeted use cases. - Collaborate on multimodal experiments (text image audio) and implement training/evaluation pipelines. - Implement data preprocessing, augmentation, and basic feature engineering for model inputs. - Run experiments, design evaluation metrics, perform ablations, log results, and iterate on models. - Optimize inference and memory footprint for models (quantization, batching, basic distillation). - Contribute to model training pipelines, scripting, and reproducible experiments. - Work with cross-functional teams to prepare prototypes for deployment. - Write clear documentation, present technical results to the team, participate in code reviews, and share knowledge. - Continuously learn by reading papers, trying new tools, and bringing fresh ideas into projects. Qualifications required for this role include: - Strong Python programming skills and familiarity with ML tooling such as numpy, pandas, and scikit-learn. - Hands-on experience (2+ years) with PyTorch and/or TensorFlow for model development and fine-tuning. - Solid grounding in classical ML & DL techniques including supervised/unsupervised learning, optimization, CNNs, RNNs/LSTMs, and Transformers. - Practical experience fine-tuning open models like Hugging Face Transformers, LLaMA family, BLOOM, Mistral, or similar. - Familiarity with PEFT approaches (LoRA, adapters, QLoRA basics) and efficiency techniques like mixed precision and model quantization. - Exposure to at least one cloud ML environment (GCP Vertex AI, AWS SageMaker, or Azure AI) for training or deployment tasks. - Good communication skills for documenting experiments and collaborating with product/infra teams. Preferred skills that would be highly desirable include: - Experience with multimodal training pipelines or cross-modal loss functions. - Familiarity with MLOps concepts and tools like DeepSpeed, Accelerate, Ray, or similar distributed/efficiency libraries. - Knowledge of LangGraph / Autogen / CrewAI or interest in agentic systems. - Experience with BigQuery / Synapse or data warehousing for analytics. - Awareness of AI safety and responsible-AI best practices. In this role, you will have the opportunity for rapid exposure to state-of-the-art generative AI and real production use cases. Additionally, you will experience a mix of research mindset and product delivery, making it ideal for individuals looking to grow into senior/researcher roles. You will receive mentor support from senior researchers and have opportunities to publish or demo prototypes. As an AI Engineer at our company in Kochi, you will play a vital role in designing, prototyping, and delivering generative AI capabilities. Your responsibilities will include working on practical research, building POCs, fine-tuning open models, contributing to multimodal experiments, and taking solutions towards production. Here is an overview of what you will do: - Build and evaluate prototypes / POCs for generative AI features and ideas. - Fine-tune and adapt open-source LLMs and smaller generative models for targeted use cases. - Collaborate on multimodal experiments (text image audio) and implement training/evaluation pipelines. - Implement data preprocessing, augmentation, and basic feature engineering for model inputs. - Run experiments, design evaluation metrics, perform ablations, log results, and iterate on models. - Optimize inference and memory footprint for models (quantization, batching, basic distillation). - Contribute to model training pipelines, scripting, and reproducible experiments. - Work with cross-functional teams to prepare prototypes for deployment. - Write clear documentation, present technical results to the team, participate in code reviews, and share knowledge. - Continuously learn by reading papers, trying new tools, and bringing fresh ideas into projects. Qualifications required for this role include: - Strong Python programming skills and familiarity with ML tooling such as numpy, pandas, and scikit-learn. - Hands-on experience (2+ years) with PyTorch and/or TensorFlow for model development and fine-tuning. - Solid grounding in classical ML & DL techniques including supervised/unsupervised learning, optimization, CNNs, RNNs/LSTMs, and Transformers. - Practical experience fine-tuning open models like Hugging Face Transformers, LLaMA family, BLOOM, Mistral, or similar. - Familiarity with PEFT approaches (LoRA, adapters, QLoRA basics) and efficiency techniques like mixed pr