Table of Contents
Learn about nnj
Overview
Jacobian in a neural network
Efficient implementation
Getting Started
Ok, but why?
Why to use nnj?
Why not Jax?
Comparison
Python API
Abstract Jacobian (diagonal)
Abstract Jacobian
Linear
MaxPool2d
ReLU
Sequential
Sigmoid
Sinusoidal
Softplus
Tanh
TruncExp
Upsample
Docs
>
Why to use nnj?
Shortcuts
Why to use nnj?
¶
Why not Jax?
¶
Comparison
¶
Why to use nnj?
Why not Jax?
Comparison
