flax
ae74bb3a - Clarifies usage of using jax.jit in Module.init()

Commit

5 years ago

Clarifies usage of using jax.jit in Module.init()

References

#910 - Clarifies usage of jit on Module.init

Author

marcvanzee

marcvanzee

Committer

marcvanzee

marcvanzee

Parents

Loading