Francois Chollet 2fa38c1071 Fix JAX trainer correctness issue and add test		2023-04-19 13:50:22 -07:00
..
core	Fix JAX trainer correctness issue and add test	2023-04-19 13:50:22 -07:00
regularization	Add Dropout layer.	2023-04-18 19:22:13 -07:00
__init__.py	Add Dropout layer.	2023-04-18 19:22:13 -07:00
input_spec.py	Add some numpy ops (#1)	2023-04-12 11:31:58 -07:00
layer_test.py	Switch TF trainer to a setup that enables future step fusing	2023-04-19 11:09:43 -07:00
layer.py	Add numpy module in jax/ and tensorflow/ (#13)	2023-04-18 18:45:30 -07:00
layers_utils.py	Initial draft of the codebase.	2023-04-09 12:21:45 -07:00