a333b9bd1b
* Adds unit normalization and tests * Adds layer normalization and initial tests * Fixes formatting in docstrings * Fix type issues for JAX * Fix nits |
||
---|---|---|
keras_core | ||
shell | ||
.gitignore | ||
demo_custom_jax_workflow.py | ||
demo_custom_layer_backend_agnostic.py | ||
demo_custom_tf_workflow.py | ||
demo_functional.py | ||
demo_subclass.py | ||
jax_integration_test.py | ||
jax_training_scratchpad.py | ||
LICENSE | ||
pyproject.toml | ||
requirements.txt | ||
setup.cfg | ||
setup.py | ||
tf_integration_test.py |