Software Engineer - Python (AI)

2 - 6 years

0 Lacs

Posted:6 days ago| Platform: Shine logo

Apply

Work Mode

On-site

Job Type

Full Time

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

Mock Interview

Practice Video Interview with JobPe AI

Start Job-Specific Interview
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

coding practice

Enhance Your Skills

Practice coding challenges to boost your skills

Start Practicing Now

RecommendedJobs for You