Skip to main content

NVIDIA NeMo Agent Toolkit

Project description

NVIDIA NeMo Agent Toolkit

NVIDIA NeMo Agent Toolkit

License: Apache 2.0 GitHub Release PyPI version GitHub issues GitHub pull requests GitHub Repo stars GitHub forks Ask DeepWiki Open in Colab

NVIDIA NeMo Agent Toolkit adds intelligence to AI agents across any framework—enhancing speed, accuracy, and decision-making through enterprise-grade instrumentation, observability, and continuous learning.

🔥 New Features

  • Dynamo Runtime Intelligence: Automatically infer per-request latency sensitivity from agent profiles and apply runtime hints for cache control, load-aware routing, and priority-aware serving.
  • Agent Performance Primitives (APP): Introduce framework-agnostic performance primitives that accelerate graph-based agent frameworks such as LangChain, CrewAI, and Agno with parallel execution, speculative branching, and node-level priority routing.
  • LangSmith Native Integration: Observe end-to-end agent execution with native LangSmith tracing, run evaluation experiments, compare outcomes, and manage prompt versions across development and production workflows.
  • FastMCP Workflow Publishing: Publish NeMo Agent Toolkit workflows as MCP servers using the FastMCP server runtime to simplify MCP-native deployment and integration.
  • Migration notice: 1.5.0 simplifies package installation and dependency management. See the Migration Guide.

✨ Key Features

  • 🛠️ Building Agents: Accelerate your agent development with tools that make it easier to get your agent into production.
    • 🧩 Framework Agnostic: Work side-by-side with agentic frameworks to add the instrumentation necessary for observing, profiling, and optimizing your agents. Use the toolkit with popular frameworks such as LangChain, LlamaIndex, CrewAI, Microsoft Semantic Kernel, and Google ADK, as well as custom enterprise agentic frameworks and simple Python agents.
    • 🔁 Reusability: Build components once and use them multiple times to maximize the value from development effort.
    • Customization: Start with a pre-built agent, tool, or workflow, and customize it to your needs.
    • 💬 Built-In User Interface: Use the NeMo Agent Toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
  • 📈 Agent Insights: Utilize NeMo Agent Toolkit instrumentation to better understand how your agents function at runtime.
    • 📊 Profiling: Profile entire workflows from the agent level all the way down to individual tokens to identify bottlenecks, analyze token efficiency, and guide developers in optimizing their agents.
    • 🔎 Observability: Track performance, trace execution flows, and gain insights into your agent behaviors in production.
  • 🚀 Agent Optimization: Improve your agent's quality, accuracy, and performance with a suite of tools for all phases of the agent lifecycle.
  • 🔌 Protocol Support: Integrate with common protocols used to build agents.

With NeMo Agent Toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.

🚀 Installation

Before you begin using NeMo Agent Toolkit, ensure that you have Python 3.11, 3.12, or 3.13 installed on your system.

[!NOTE] For users who want to run the examples, it's required to clone the repository and install from source to get the necessary files required to run the examples. Please refer to the Examples documentation for more information.

To install the latest stable version of NeMo Agent Toolkit from PyPI, run the following command:

pip install nvidia-nat

NeMo Agent Toolkit has many optional dependencies that can be installed with the core package. Optional dependencies are grouped by framework. For example, to install the LangChain/LangGraph plugin, run the following:

pip install "nvidia-nat[langchain]"

Detailed installation instructions, including the full list of optional dependencies and their conflicts, can be found in the Installation Guide.

🌟 Hello World Example

Before getting started, it's possible to run this simple workflow and many other examples in Google Colab with no setup. Click here to open the introduction notebook: Open in Colab.

  1. Ensure you have set the NVIDIA_API_KEY environment variable to allow the example to use NVIDIA NIMs. An API key can be obtained by visiting build.nvidia.com and creating an account.

    export NVIDIA_API_KEY=<your_api_key>
    
  2. Create the NeMo Agent Toolkit workflow configuration file. This file will define the agents, tools, and workflows that will be used in the example. Save the following as workflow.yml:

    functions:
       # Add a tool to search wikipedia
       wikipedia_search:
          _type: wiki_search
          max_results: 2
    
    llms:
       # Tell NeMo Agent Toolkit which LLM to use for the agent
       nim_llm:
          _type: nim
          model_name: nvidia/nemotron-3-nano-30b-a3b
          temperature: 0.0
          chat_template_kwargs:
             enable_thinking: false
    
    workflow:
       # Use an agent that 'reasons' and 'acts'
       _type: react_agent
       # Give it access to our wikipedia search tool
       tool_names: [wikipedia_search]
       # Tell it which LLM to use
       llm_name: nim_llm
       # Make it verbose
       verbose: true
       # Retry up to 3 times
       parse_agent_response_max_retries: 3
    
  3. Run the Hello World example using the nat CLI and the workflow.yml file.

    nat run --config_file workflow.yml --input "List five subspecies of Aardvarks"
    

    This will run the workflow and output the results to the console.

    Workflow Result:
    ['Here are five subspecies of Aardvarks:\n\n1. Orycteropus afer afer (Southern aardvark)\n2. O. a. adametzi  Grote, 1921 (Western aardvark)\n3. O. a. aethiopicus  Sundevall, 1843\n4. O. a. angolensis  Zukowsky & Haltenorth, 1957\n5. O. a. erikssoni  Lönnberg, 1906']
    

📚 Additional Resources

🛣️ Roadmap

  • Automatic Reinforcement Learning (RL) to fine-tune LLMs for a specific agent.
  • Integration with NVIDIA Dynamo to reduce LLM latency at scale.
  • Improve agent throughput with KV-Cache optimization.
  • Improved, standalone evaluation harness and migration to ATIF for trajectory format.
  • Support for additional programming languages (TypeScript, Rust, Go, WASM) with compiled libraries.
  • Phasing out wrapping architecture to ease onboarding for more agents.
  • Support for adding skills and sandboxes to existing agents.
  • MCP authentication improvements.
  • Improved memory interface to support self-improving agents.

💬 Feedback

We would love to hear from you! Please file an issue on GitHub if you have any feedback or feature requests.

🤝 Acknowledgements

We would like to thank the following groups for their contribution to the toolkit:

  • Synopsys
    • Google ADK framework support.
    • Microsoft AutoGen framework support.
  • W&B Weave Team
    • Contributions to the evaluation and telemetry system.

In addition, we would like to thank the following open source projects that made NeMo Agent Toolkit possible:

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

nvidia_nat-1.5.0a20260316-py3-none-any.whl (52.7 kB view details)

Uploaded Python 3

File details

Details for the file nvidia_nat-1.5.0a20260316-py3-none-any.whl.

File metadata

File hashes

Hashes for nvidia_nat-1.5.0a20260316-py3-none-any.whl
Algorithm Hash digest
SHA256 b99ad034a5d4490b1dc6411ec9c52341369d8d17802e8c3c920e212aa83c5d99
MD5 2b12fb67d6baa5d8b5ab475f6087372f
BLAKE2b-256 25da6e555ee1b884aeff32b5081ff104b8ab33669944c942ca87d94a5b64b6ad

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page