Skip to main content

Ctrl+K

Warp-NN 0.4.1

GitHub
PyPI

Warp-NN 0.4.1

GitHub
PyPI

Table of Contents

User Guide

API Reference
- Activations
  - CELU
  - ELU
  - GELU
  - Hard Sigmoid
  - Hard Swish
  - Leaky ReLU
  - Log-Softmax
  - Mish
  - PReLU
  - ReLU
  - SELU
  - Shrink
  - Sigmoid
  - Softmax
  - Softplus
  - Softsign
  - Swish
  - Threshold
- Initializers
- Layers
  - AvgPool1D
  - AvgPool2D
  - BatchNorm
  - Buffer
  - Conv1D
  - Conv2D
  - Dropout
  - Flatten
  - GlobalAvgPool
  - GlobalMaxPool
  - GroupNorm
  - GRU
  - GRUCell
  - Identity
  - InstanceNorm
  - LayerNorm
  - Linear
  - LSTM
  - LSTMCell
  - MaxPool1D
  - MaxPool2D
  - Module
  - Parameter
  - RMSNorm
  - RNN
  - RNNCell
  - Sequential
- Operators
  - Abs
  - Acos
  - Acosh
  - Add
  - Asin
  - Asinh
  - Atan
  - Atanh
  - BitwiseAnd
  - BitwiseNot
  - BitwiseOr
  - BitwiseXor
  - Ceil
  - Clip
  - Cos
  - Cosh
  - Div
  - Erf
  - Exp
  - Floor
  - Log
  - Max
  - Min
  - Mul
  - Neg
  - Pow
  - Reciprocal
  - Round
  - Sign
  - Sin
  - Sinh
  - Sqrt
  - Sub
  - Tan
  - Tanh
- Optimizers
  - Adam
  - SGD
- Runtime
  - OnnxRuntime
- Utils
  - Kernel Configuration
  - Device
  - Graph
  - 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.