JazzX AI

JazzX AI

Senior Staff AI Engineer

Los Altos, California, · Staff+

Sponsorship not specifiedDetected 50 days ago
PythonJavaC++AlgorithmsKubernetesPlatform EngineeringMachine LearningTensorFlowPyTorchNLPLLMsA/B TestingRoboticsResearchLeadershipCommunicationCollaborationMentoring

About the role

  • We are seeking an experienced AI Engineer with deep expertise in Reinforcement Learning (RL) to join our team as a Senior Staff Architect.
  • In this role, you will be responsible for shaping the vision, architecture, and technical execution of RL-driven AI reasoning models and systems that power next-generation enterprise AGI platform.
  • Ultimately, your work will be critical in bridging research and production, ensuring that the latest RL advancements translate into reliable, impactful, and enterprise-ready AI solutions.

Responsibilities

  • Scalability & Infrastructure: Design distributed training systems, leverage cloud-native infrastructure, and optimize for performance, reproducibility, and cost-efficiency.
  • review designs and code with a focus on scalability, robustness, and clarity.
  • Collaboration: Partner with product, data, and platform teams to align RL solutions with strategic business goals and integrate them into production systems.
  • Leadership & Mentorship: Provide technical leadership to AI engineers and researchers; mentor junior team members; review designs and code with a focus on scalability, robustness, and clarity.
  • From here, we're building the backbone for enterprise intelligence across industries.
  • JazzX AI is defining the future of enterprise work-by building AI-native digital workers that actually get the job done.
  • What draws people to JazzX AI is the opportunity to build something truly foundational.
  • At JazzX AI, you'll help build a platform that redefines how work happens in the AGI era.
  • If you're motivated by building systems that matter - systems that combine intelligence with responsibility - JazzX AI offers the chance to do that work at a deeper level, with real ownership and lasting impact.

Requirements

  • Experience with test-time compute optimization techniques, including inference-time search, chain-of-thought reasoning, and adaptive computation strategies for improving model performance during deployment.
  • Advanced software engineering skills in Python, C++, or Java, with deep expertise in ML frameworks such as TensorFlow, PyTorch, JAX, or Ray RLlib.
  • Hands-on experience with distributed training infrastructure (Kubernetes, GPU/TPU clusters, and cloud ML platforms).

Nice to have

  • Experience leading enterprise AI adoption and guiding organizational strategy for RL powered systems.
  • Contributions to open-source RL frameworks or publications in top-tier conferences (NeurIPS, AISTATS, ICML, ICLR, AAAI).
  • Background in safety, alignment, or explainability of RL agents.
  • Why Join JazzX AI:
  • This is not about experimenting with AI features
  • it's about turning the promise of AGI into practical, auditable, and human-aligned systems that enterprises can trust and rely on at scale.
  • The work here directly impacts productivity, efficiency, and decision-making in complex, real-world environments.
  • Headquartered in Los Altos, CA.

Skills

  • Define frameworks for benchmarking, continuous evaluation, and feedback-driven improvements in deployed RL models.

Compensation

  • SAIGroup's portfolio serves 2,000+ global enterprise customers, generates nearly $800M in annual revenue, and employs 4,000+ people worldwide — providing JazzX AI with long-term capital, deep operating expertise, and access to real-world en

Benefits

  • Vision: Enterprises operating on institutional intelligence-governed, self-improving, and scalable beyond individual expertise.
  • JazzX AI brings together vision, deep technology, and purpose.
  • Architecture & Design: Define and drive the end-to-end architecture for reinforcement learning-based systems, including training pipelines, simulation environments, reward shaping, and model serving.

Company info

  • https://www.linkedin.com/company/jazzx-ai
  • We are at the beginning of a new era - one where enterprise systems will no longer be static tools, but dynamic collaborators that learn, reason, and evolve alongside the people who use them.
Apply directly at JazzX AI →Create a free account for alerts like thisView JazzX AI immigration profile

This listing is sourced directly from JazzX AI's careers page and normalized into a canonical job model.