keras/keras_core/backend/jax
2023-05-13 22:54:25 -07:00
..
__init__.py Refactor backend to remove logic from __init__ 2023-05-13 18:05:07 -07:00
core.py Refactor backend to remove logic from __init__ 2023-05-13 18:05:07 -07:00
image.py Merge branch 'main' of github.com:keras-team/keras-core 2023-05-05 22:09:34 -07:00
math.py Added Softmax activation layer (#115) 2023-05-09 16:03:15 -07:00
nn.py Fix jax bug + annoying TF warning 2023-05-13 22:54:25 -07:00
numpy.py Adds CategoryEncoding layer, bincount op, and tests (#161) 2023-05-14 00:07:43 +00:00
random.py fix documentation in the random module (#141) 2023-05-11 22:03:40 +05:30
rnn.py Refactor backend to remove logic from __init__ 2023-05-13 18:05:07 -07:00
trainer.py Add steps_per_execution to jax backend (#165) 2023-05-13 17:15:05 -07:00