flax
edad3e9e - fix nn.value_and_grad by implementing directly in core

Commit

2 years ago

fix nn.value_and_grad by implementing directly in core

References

#3479 - fix nn.value_and_grad by implementing directly in core

Author

levskaya

levskaya

Parents

Loading