flax
1a78acb5 - [nnx] use jax-like transforms API in nnx_basics

Commit

1 year ago

[nnx] use jax-like transforms API in nnx_basics

References

#4155 - [nnx] use jax-style transforms API in nnx_basics

Author

cgarciae

cgarciae

Committer

cgarciae

cgarciae

Parents

Loading