Francois Chollet 4bf04920f2 Fix jax jit_compile setting		2023-04-19 14:35:28 -07:00
..
activations	Add Functional model draft.	2023-04-12 14:27:30 -07:00
backend	Fix jax jit_compile setting	2023-04-19 14:35:28 -07:00
callbacks	Make progbar logger work	2023-04-17 16:11:30 -07:00
constraints	Export initializers	2023-04-19 12:52:58 -07:00
initializers	Export initializers	2023-04-19 12:52:58 -07:00
layers	Fix JAX trainer correctness issue and add test	2023-04-19 13:50:22 -07:00
losses	Fix indent	2023-04-18 21:18:55 -07:00
metrics	Add fully operational JAX trainer.	2023-04-18 14:49:38 -07:00
models	Add fully operational JAX trainer.	2023-04-18 14:49:38 -07:00
operations	Add numpy module in jax/ and tensorflow/ (#13)	2023-04-18 18:45:30 -07:00
optimizers	Add fully operational JAX trainer.	2023-04-18 14:49:38 -07:00
regularizers	Export initializers	2023-04-19 12:52:58 -07:00
saving	Initial draft of the codebase.	2023-04-09 12:21:45 -07:00
testing	Added MeanAbsoluteError and started testing for losses. (#11)	2023-04-19 00:18:20 -04:00
trainers	Fix JAX trainer correctness issue and add test	2023-04-19 13:50:22 -07:00
utils	Add code stats util	2023-04-16 14:54:32 -07:00
__init__.py	Add demos	2023-04-18 15:46:57 -07:00
api_export.py	Add some numpy ops (#1)	2023-04-12 11:31:58 -07:00