DifferentiableFlatten.jl

Zygote-differentiable way of flattening and unflattening data structures
Author JuliaNonconvex
Popularity
2 Stars
Updated Last
11 Months Ago
Started In
October 2022

DifferentiableFlatten

Actions Status codecov

This package includes an implementation of a flatten function which flattens all data structures to vectors and returns an unflattenning function to back to the original data structure. This was originally part of NonconvexCore.jl.