Senior Software Engineer, AI Frameworks

We are seeking a Senior Software Engineer to drive integration of the NVIDIA Grove project within Dynamo and across a set of leading open-source AI frameworks. In this role, you will develop production-grade software enabling Grove capabilities to be adopted, scaled, and operated smoothly. In this role, you will build production-grade software that enables seamless adoption, scaling, and operation of Grove capabilities across environments such as Dynamo, llm-d, Ray, PyTorch, and other emerging frameworks in the AI ecosystem. You will collaborate across engineering teams and the open-source community to deliver robust integrations, reference implementations, and developer-focused tooling.

What you'll be doing:

• Design and implement end-to-end integrations of Grove with open-source AI frameworks (e.g., Dynamo, llm-d, Ray, PyTorch, and related ecosystem projects).
• Build and maintain adapters, plugins, operators, and/or runtime components that enable Grove features to work smoothly across training and inference stacks.
• Partner with framework owners to upstream changes, contribute patches, and ensure long-term maintainability of integrations.
• Develop reference workflows, sample apps, and best-practice guides that accelerate adoption by users and partners.
• Optimize performance, scalability, and reliability for distributed training/inference, including multi-node and multi-GPU environments.
• Improve observability and operational readiness (metrics, logging, tracing, debugging tools) for Kubernetes-based deployments.
• Participate in technical design reviews, define APIs/contracts, and ensure compatibility across versions of frameworks and dependencies.
• Diagnose complex issues spanning containers, networking, scheduling, CUDA/GPU utilization, and framework runtime behavior.

What we need to see:

…