Metadata-Version: 2.4
Name: warp-nn
Version: 0.2.0
Summary: Neural network library for NVIDIA Warp
Author: NVIDIA Corporation
License: Apache-2.0
Project-URL: Documentation, https://nvidia.github.io/warp-nn
Project-URL: Repository, https://github.com/NVIDIA/warp-nn
Project-URL: Issues, https://github.com/NVIDIA/warp-nn/issues
Project-URL: Changelog, https://github.com/NVIDIA/warp-nn/blob/main/CHANGELOG.md
Keywords: machine-learning,neural-network,nvidia,warp
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE.md
Requires-Dist: warp-lang>=1.12.0
Provides-Extra: tests
Requires-Dist: hypothesis; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: pytest-cov; extra == "tests"
Requires-Dist: torch; extra == "tests"
Provides-Extra: docs
Requires-Dist: myst-parser; extra == "docs"
Requires-Dist: nvidia-sphinx-theme; extra == "docs"
Requires-Dist: sphinx-autobuild; extra == "docs"
Requires-Dist: sphinx-copybutton; extra == "docs"
Dynamic: license-file

[![PyPI version](https://badge.fury.io/py/warp-nn.svg)](https://badge.fury.io/py/warp-nn)
[![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)

# Warp-NN: CUDA Graphable Neural Networks for NVIDIA Warp

**[Documentation](https://nvidia.github.io/warp-nn/latest)** | [Changelog](https://github.com/NVIDIA/warp-nn/blob/main/CHANGELOG.md)

Warp-NN is a Warp-native Python library for building and training neural networks for Physical AI workflows.
It is designed for compact neural network components that run directly within Warp-based simulation,
robotics, control, and differentiable computing pipelines.
It is not intended to be a general-purpose replacement for PyTorch, JAX, or other ML frameworks.

> **Disclaimer:**
> Warp-NN is not part of the `warp-lang` package, and it is not maintained by the [NVIDIA Warp](https://nvidia.github.io/warp) core team.
> It is a Warp ecosystem library maintained by [@Toni-SM](https://github.com/Toni-SM) / [NVIDIA Isaac Sim](https://github.com/isaac-sim).
> Issues, releases, roadmap, and support are managed by the maintainers of this repository.

## Installation

The easiest way to install Warp-NN is from [PyPI](https://pypi.org/project/warp-nn).
Refer to the *Installation* section in docs for more details.

```bash
pip install warp-nn
```

## Support

Questions and discussions can be opened on [GitHub Discussions](https://github.com/NVIDIA/warp-nn/discussions).

Problems, issues, and feature requests can be opened on [GitHub Issues](https://github.com/NVIDIA/warp-nn/issues).

## Contributing

Contributions and pull requests from the community are welcome.
Please see the [Contribution Guide](https://github.com/NVIDIA/warp-nn/blob/main/CONTRIBUTING.md)
for more information on contributing to the development of Warp-NN.

## License

Warp-NN is provided under the Apache License, Version 2.0.
Please see [LICENSE.md](https://github.com/NVIDIA/warp-nn/blob/main/LICENSE.md) for full license text.

This project will download and install additional third-party open source software projects.
Review the license terms of these open source projects before use.
