Francois Chollet 14a10bcd1f Add bulk of serialization logic.		2023-04-21 10:00:32 -07:00
..
data_adapters	Rename data_adapter_utils	2023-04-18 21:11:21 -07:00
__init__.py	Start work on TF Trainer and DataAdapters.	2023-04-16 20:57:14 -07:00
compile_utils_test.py	Misc fixes + add support for JAX array input	2023-04-17 17:47:22 -07:00
compile_utils.py	Add JAX evaluate flow.	2023-04-19 18:37:25 -07:00
epoch_iterator_test.py	Add EpochIterator abstraction	2023-04-17 14:56:16 -07:00
epoch_iterator.py	Switch TF trainer to a setup that enables future step fusing	2023-04-19 11:09:43 -07:00
trainer_test.py	Add bulk of serialization logic.	2023-04-21 10:00:32 -07:00
trainer.py	Remove debug statement	2023-04-19 18:44:44 -07:00