Metadata-Version: 2.4
Name: warp-nn
Version: 0.1.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

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

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

Warp-NN is a Python library for building and training neural networks using [NVIDIA Warp](https://nvidia.github.io/warp).
It enables end-to-end GPU-accelerated neural network implementation for Warp-based applications,
without the need for third-party libraries such as PyTorch or JAX.

## 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.
