Warp-NN is a Python library for building and training neural networks using NVIDIA 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.
The easiest way to install Warp-NN is from PyPI. Refer to the Installation section in docs for more details.
pip install warp-nnQuestions and discussions can be opened on GitHub Discussions.
Problems, issues, and feature requests can be opened on GitHub Issues.
Contributions and pull requests from the community are welcome. Please see the Contribution Guide for more information on contributing to the development of Warp-NN.
Warp-NN is provided under the Apache License, Version 2.0. Please see 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.
