Search — nnj 0.0.0 documentation

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 >
Search

Shortcuts

© Copyright 2023, Marco Miani and Frederik Warburg.

Built with Sphinx using a theme provided by Read the Docs.