Skip to main content

Ctrl+K

Warp-NN 0.1.0

GitHub
PyPI

Warp-NN 0.1.0

GitHub
PyPI

Table of Contents

User Guide

API Reference
- Activations
  - ELU
  - Leaky ReLU
  - ReLU
  - SELU
  - Sigmoid
  - Soft-plus
  - Soft-sign
  - Tanh
- Initializers
- Layers
  - Conv1D
  - Conv2D
  - GRUCell
  - Linear
  - LSTMCell
  - Module
  - Parameter
  - RNNCell
  - Sequential
- Optimizers
  - Adam
  - SGD
- Utils
  - Device
  - Graph
  - Kernel Configuration
  - Logging
  - Ops

Examples

User Guide

User Guide#

previous

Warp-NN: CUDA Graphable Neural Networks for NVIDIA Warp

next

Installation

Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA.