InvertibleNetworks.jl

A Julia framework for invertible neural networks

MIT License

Stars
151
Committers
17

Bot releases are hidden (Show)

InvertibleNetworks.jl - v2.2.6

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

InvertibleNetworks.jl - Chainrules

Published by mloubout over 3 years ago

  • Support for chainrules and Zygote
  • Fix GPU support
  • Make layers/network size independent to follow general conventions and generalizations.
InvertibleNetworks.jl - GPU support

Published by mloubout over 3 years ago

Fixes GPU support:

  • Fix dimensionality manipulation views
  • Fix loss function input types
  • Add GPU supported Haar wavelet squeeze/unsqueeze

Simplifies and fix bugs in Hyperbolic layer
Add documentation

InvertibleNetworks.jl - Jacobians and adjoint Jacobians of layers and networks

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

InvertibleNetworks.jl -

Published by mloubout almost 4 years ago

Support of Jacobians and adjoint of layers and networks.

InvertibleNetworks.jl - First release

Published by mloubout over 4 years ago

Package Rankings
Top 16.55% on Juliahub.com
Badges
Extracted from project README's
CI DOI
Related Projects