flax
eecf36fc - unify variable_in_axes and variable_out_axes

Commit

5 years ago

unify variable_in_axes and variable_out_axes

References

#451 - unify variable_in_axes and variable_out_axes

Author

jheek

jheek

Committer

jheek

jheek

Parents

Loading