Senior Staff Engineer - AI Engineering

Role overview

Senior Staff Engineer - AI Engineering at Lattice Semiconductor India

AI Strategy and Leadership:

Responsibilities

• Define and execute AI strategies focused on RAG-based retrieval, code generation, and AI-assisted software engineering
• Work with stakeholders to align AI capabilities with business objectives and software development needs
• Research and integrate cutting-edge LLMs and autonomous AI agent architecture into development processes.
• RAG & Agentic AI Development:
• Develop RAG pipelines that enhance AI‘s ability to retrieve relevant knowledge and generate context-aware responses.
• Build and optimize agentic AI systems that can interact with APIs, databases, and development environments (such as LangChain, OpenAI APIs, etc.)
• Implement AI-powered search, chatbots, and decision-support tools for software engineers.
• Fine-tune LLMs (GPT, Llama, Mistral, Claude, Gemini etc.) for domain-specific applications.
• Optimize retrieval mechanisms to enhance response accuracy, grounding AI outputs in real-world data
• Code generation & Test case Automation:
• Leverage LLMs to generate high-quality, production-ready code
• Develop AI-driven test case generation tools that automatically create and validate unit tests, integration tests, and regression tests
• Integrate AI-driven code assistants and programming agents into IDE and CI/CD workflows
• Optimize prompt engineering and fine-tuning strategies for LLMs to improve code quality and efficiency
• MLOps & Scalable AI Systems:
• Architect and deploy scalable AI models and retrieval pipelines using cloud-based MLOps pipelines (AWS/GCP/Azure, Docker, Kubernetes)
• Optimize LLMs for real-time AI inferencing , ensuring low latency and high-performance AI solutions.
• Collaboration:
• Work cross-functionally with product teams, software engineers, and business stakeholders to integrate AI solutions into products.
• Mentorship:
• Guide and mentor a team of 3-5 AI engineers in LLM fine-tuning, retrieval augmentation, and autonomous AI agents.
• Establish best practices for AI-assisted software development, secure AI integration, and bias mitigation.
• Research & Innovation:
• Commitment to staying updated with the latest AI and machine learning research and advancements .
• Ability to think creatively and propose innovative solutions to complex problems.
• Model Development:
• Ability to design, train, and evaluate various AI models , including LLMs and standalone models —familiarity with model training tools and frameworks like Hugging Face Trainer, Fairseq, etc .

Required qualifications

• Master's or Ph.D. in Computer Science, AI, Machine Learning, or a related field.
• 5+ years of experience in AI and machine learning, with at least 2 years of experience working on LLMs, code generation, RAG, or AI-powered automation .

Employment details

• Location: Pune [India]

Additional role details

• Lattice Overview
• There is energy here…energy you can feel crackling at any of our international locations.
• It’s an energy generated by enthusiasm for our work, for our teams, for our results, and for our customers.
• Lattice is a worldwide community of engineers, designers, and manufacturing operations specialists in partnership with world-class sales, marketing, and support teams, who are developing programmable logic solutions that are changing the industry.
• Our focus is on R&D, product innovation, and customer service, and to that focus, we bring total commitment and a keenly sharp competitive personality.
• Energy feeds on energy.
• If you flourish in a fast paced, results-oriented environment, if you want to achieve individual success within a “team first” organization, and if you believe you can contribute and succeed in a demanding yet collegial atmosphere, then Lattice may well be just what you’re looking for.
• This position requires a successful background and reference checks and satisfactory proof of your right to work in:
• India
• Lattice recognizes that employees are its greatest asset and the driving force behind success in a highly competitive, global industry.
• Lattice continually strives to provide a comprehensive compensation and benefits program to attract, retain, motivate, reward and celebrate the highest caliber employees in the industry.
• Lattice is an international, service-driven developer of innovative low cost, low power programmable design solutions.
• Our global workforce, some 1,000 strong, shares a total commitment to customer success and an unbending will to win.
• For more information about how our FPGA, CPLD and programmable power management devices help our customers unlock their innovation, visit www.latticesemi.com .
• You can also follow us via Twitter, Facebook, or RSS.
• At Lattice, we value the diversity of individuals, ideas, perspectives, insights and values, and what they bring to the workplace.
• Applications are welcome from all qualified candidates.