Repository Issue Activity (beta feature)

#5433 Add more examples in the NNX docs

#

#

#

#

#

#

#5412 nnx.LoRA.__call__ doesn't forward **kwargs to base_module, breaks models with parameterized linear layers

#

#

#4488 Failed build: CI - with JAX nightly

#

#5194 Improvement on the Training API

#

#

#

#

#

#

#

#

#

#

#

#