You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 4de57ccf01 Add weight init scheme that's closer to JAX impl		4 years ago
..
data	Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.	4 years ago
loss	…
models	Add weight init scheme that's closer to JAX impl	4 years ago
optim	…
scheduler	…
utils	…
__init__.py	…
version.py	…