A Julia framework for invertible neural networks
MIT License
Bot releases are hidden (Show)
Published by rafaelorozco 11 months ago
Summarizednetwork + docs
-summarized network structure beautifies the summary network workflow
-2 examples to docs will serve as template for many examples to come soon
Published by mloubout over 3 years ago
Published by mloubout over 3 years ago
Fixes GPU support:
Simplifies and fix bugs in Hyperbolic layer
Add documentation
Published by mloubout almost 4 years ago
Implementation of Jacobian evaluation (and adjoint thereof) for invertible networks
Main contributions:
added Jacobian functionalities for most of the layers and networks already implemented, except for layers and networks requiring a "link" function (invertible_layer_slim_additive
, invertible_layer_slim_affine
, invertible_network_irim
) and Flux blocks that requires forward-mode AD;
Jacobians can be wrapped around an abstract linear algebra framework, see /examples/jacobian/jacobian_linear_algebra.jl
;
composition functionalities to string networks/layers together, see /examples/utils/composition_of_networks.jl
Jacobians also work together with Composition.
new layers/networks: layer_resnet (as a Flux block), invertible_network_hint_multiscale (previously only had a conditional version)
utilities for Parameter
or Array{Parameter, 1}
direct manipulation, see /examples/utils/parameter_algebra.jl
Published by mloubout over 4 years ago