axon icon indicating copy to clipboard operation
axon copied to clipboard

Published 20 hours ago • verified publisher icon elixir-nx

Use layer key meta to split keys

Open seanmor5 opened this issue 2 years ago • 0 comments

Trying to align more with Flax's implementation which does not split keys but rather generates keys from layer meta

seanmor5 avatar Oct 15 '22 15:10 seanmor5