Senior Staff Engineer - AI Engineering

Location: Pune, India

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.

Job Description:

Key Responsibilities:

• AI Strategy and Leadership:
  • 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:

• Education: Master's or Ph.D. in Computer Science, AI, Machine Learning, or a related field.
• Experience: 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.

Additional Information:

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.