Staff AI Engineer

JazzX AI, a SAIGroup company, is seeking a Staff AI Research Engineer to join our team in Bengaluru, India. This hybrid role focuses on applied AI and Generative AI platforms and applications, requiring 10+ years of experience. You will play a crucial role in shaping our core AI platform, Builder Studio abstractions, and AI capabilities for enterprise applications. This is a hands-on individual contributor position where you will design and validate the consumption of foundation models, reasoning systems, and agentic workflows by both technical and non-technical users.

Your work will directly influence how AI systems are composed and deployed via low-code/no-code interfaces, how customer-facing AI applications perform in production, and how JazzX balances power, safety, observability, and usability across its SaaS and PaaS offerings.

What You Will Do

Applied AI Research & Experimentation

• Explore and evaluate foundation models, reasoning models, multimodal models, and agentic workflows for enterprise use cases.
• Design experiments to understand model capabilities, limits, failure modes, and trade-offs.
• Assess how advanced AI techniques (RAG, dynamic prompting, RL, knowledge graphs, ontologies) can be safely abstracted into a Builder Studio experience.

Platform & System Architecture

Design core AI platform components for:

• Model orchestration and routing
• Agent execution and state management
• Context, memory, and retrieval pipelines
• Evaluation, observability, and guardrails

• Define architectural patterns that work across platform APIs, low-code builders, and end-user applications.

Builder Studio Enablement (Low-Code / No-Code)

• Help design declarative abstractions and primitives that allow AI workflows to be composed visually or via configuration.
• Ensure that low-code experiences do not compromise correctness, safety, or performance.
• Work closely with UX, product, and platform teams to translate complex AI systems into intuitive building blocks.

First-Party Application Development

• Collaborate on building and evolving customer-facing enterprise AI applications using the JazzX platform and Builder Studio.
• Use these applications as real-world feedback loops to validate platform design decisions.
• Drive a strong dogfooding culture, ensuring the platform is battle-tested at scale.

Evaluation, Reliability & Safety

• Design and implement evaluation pipelines to continuously measure system quality.
• Develop strategies for hallucination mitigation, grounding, human-in-the-loop workflows, and deterministic guardrails.
• Ensure AI systems behave predictably in enterprise environments with high trust requirements.

Technical Leadership & Influence

• Serve as a Staff-level technical leader, influencing architecture and long-term platform direction.
• Mentor engineers and scientists on applied AI system design and AI-native engineering practices.
• Contribute to technical standards, RFCs, and cross-team architectural decisions.

What We're Looking For

Required Qualifications

• 10+ years of experience in applied AI, AI engineering, or AI-driven platform development.
• Strong hands-on experience building production-grade AI systems.
• Expert-level proficiency in Python, with a deep understanding of its ecosystem for building, scaling, and deploying production-grade AI services and orchestration layers.
• Deep experience working with AI foundation models (LLMs, reasoning models, multimodal models).
• Solid understanding of AI agents, orchestration frameworks, and evaluation methodologies.
• Experience with frameworks such as OpenAI SDK, LangChain, AutoGen, CrewAI, or equivalent.
• Knowledge of or experience with reinforcement learning, ontologies, knowledge graphs, or hybrid AI approaches.
• Ph.D. or MS in Computer Science, AI, or a related field is preferred.
• Strong analytical and systems-thinking skills.

What Sets You Apart

• You enjoy working across layers—from models to platforms to end-user applications.
• You can translate cutting-edge AI capabilities into safe, usable, and scalable abstractions.
• You thrive in ambiguous environments and can define both the problem and the solution.
• You lead through technical influence and execution, not hierarchy.
• You care deeply about production readiness, reliability, and user trust.

Join JazzX AI to build something truly foundational in the AGI era. We offer the opportunity to turn the promise of AGI into practical, auditable, and human-aligned systems that enterprises can trust. If you are motivated by building systems that matter and combine intelligence with responsibility, JazzX AI offers a chance to do that work with real ownership and lasting impact.