Skip to content

Add functool.wraps() annotation to flax.nn.jit. #13430

Add functool.wraps() annotation to flax.nn.jit.

Add functool.wraps() annotation to flax.nn.jit. #13430