keras/keras_core/backend/jax
2023-06-12 10:40:40 -07:00
..
__init__.py Add the stop_gradient function to keras_core.backend (#282) 2023-06-08 04:30:26 +00:00
core.py Add the stop_gradient function to keras_core.backend (#282) 2023-06-08 04:30:26 +00:00
image.py Merge branch 'main' of github.com:keras-team/keras-core 2023-05-05 22:09:34 -07:00
layer.py Extensive refactor of tracking logic. 2023-06-05 18:39:26 -07:00
math.py Add QR op and fix orthogonal initializer. 2023-06-10 11:58:13 -07:00
nn.py Add conv ops to torch backend (#206) 2023-05-23 15:46:26 -07:00
numpy.py Merge branch 'main' of github.com:keras-team/keras-core 2023-06-10 11:58:52 -07:00
random.py Fix dropout when noise_shape contains None (#264) 2023-06-05 16:40:19 +00:00
rnn.py Enable training tests and fix a range of bugs 2023-05-19 11:40:17 -07:00
trainer.py Fix Aurelien's new issue 2023-06-12 10:40:40 -07:00