Role : Associate Architect - Machine Learning (Gen AI)
Experience : 6 to 8 Years
Location : Bangalore / Mumbai (Hybrid)
Job Summary:
We are looking for an experienced Associate Architect - Machine Learning to join our team, focused on building Agentic AI workflows, fine-tuning Large Language Models (LLMs), performing prompt engineering, and applying related generative AI techniques.
The ideal candidate will have expertise in cutting-edge AI technologies and the ability to design, develop, and deploy AI solutions that can autonomously perform tasks with minimal human intervention.
Roles and Responsibilities:
- Agentic AI Development : Design, develop, and optimize domain adaptive agentic AI systems that helps in automating business processes
- LLM Fine-Tuning : Work with large-scale pre-trained models (like Llama, Mistral etc.) to fine-tune with techniques like PEFT, SFT and adapt them for specific applications and domains.
Evaluate and Optimize for performance, accuracy, and efficiency.
- Prompt Engineering : Design prompts with techniques like Chain of Thought, Few Shot to enhance model responses, ensuring that model outputs are aligned with use case requirements.
- AI Workflow Automation : Build end-to-end workflows for AI solutions, from data collection and preprocessing to training, deployment, and continuous improvement in production environments.
- Collaboration with Cross-functional Teams : Work closely with data scientists, software engineers, and product managers to define AI product requirements and deliver innovative solutions.
- Research & Development : Stay current with the latest research and developments in generative AI, deep learning, NLP, reinforcement learning, and related fields to ensure that the organization stays at the forefront of technology.
- Scaling and Deployment : Deploy machine learning models at scale, optimizing for latency, throughput, and robustness in production environments.
- Documentation & Reporting : Maintain clear documentation of models, workflows, and experiments, and communicate results effectively to stakeholders.
Skill Set Required:
Experience :
- Minimum 5+ years of hands-on experience in machine learning and AI engineering.
- Proven track record in working with LLMs such as Llama, Mistral and models like GPT, BERT, T5, or similar.
- Expertise in designing, fine-tuning, and deploying generative AI models and building agentic workflows.
- Strong experience in prompt engineering to optimize AI models performance.
Technical Skills :
- Proficiency in Python, TensorFlow, PyTorch, or other ML frameworks.
- Proficiency in building agentic workflows with tools like Langgraph, CrewAI, Autogen, PhiData or similar.
- Familiarity with cloud platforms (AWS, GCP, Azure) for deployment and scaling of models.
- Experience with NLP tasks, such as text classification, text generation, summarization, and question answering.
- Knowledge of reinforcement learning, multi-agent systems, or other autonomous decision-making frameworks.
- Familiarity with SDLC life cycle , data processing tools (e.g., Pandas, NumPy, etc.) and version control (e.g., Git).
Soft Skills :
- Strong problem-solving and analytical skills.
- Excellent communication and teamwork abilities to collaborate with stakeholders.
- Ability to work independently and drive projects to completion with minimal supervision.
Preferred Skills & Qualifications:
- Experience in deploying AI models at scale in production environments.
- Expertise in large-scale data processing, optimization techniques, and model deployment.