Senior Staff Engineer - AI Engineering

Overview

Senior Staff Engineer for AI Engineering at Lattice, defining and executing AI strategy around RAG-based retrieval, code generation and AI-assisted software engineering.

What you'll do

• 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.
• 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.
• 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.
• 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.
• Work cross-functionally with product teams, software engineers, and business stakeholders to integrate AI solutions into products.
• 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.
• Stay updated with the latest AI and machine learning research and advancements, and propose innovative solutions to complex problems.
• Design, train, and evaluate various AI models, including LLMs and standalone models, using tools and frameworks like Hugging Face Trainer and Fairseq.

What you'll need

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

Details

• Location: Pune, India.
• Requires a successful background and reference checks and satisfactory proof of right to work in India.