google / flax

Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.8k stars 613 forks source link

typo in nnx_basics.md #4046

Closed notnot closed 1 week ago

notnot commented 1 week ago

Replace Since Modules hold their own state methods can be called directly without the no need for a separate apply method by Since Modules hold their own state methods can be called directly without the need for a separate apply method

rajasekharporeddy commented 1 week ago

Thanks for reporting. #4047 will fix this.

notnot commented 1 week ago

Found another one...

Replace In practice this means that NNX programs can be express using imperative code by In practice this means that NNX programs can be expressed using imperative code

rajasekharporeddy commented 6 days ago

Thanks! #4050 addresses the typos in nnx_basics.md file.