Angelos Katharopoulos f5ece1d98d Tmp		2024-01-22 18:38:11 -08:00
..
layers	Tmp	2024-01-22 18:38:11 -08:00
__init__.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
losses.py	Update binary_cross_entropy function to handle both logits and probabilities (#492)	2024-01-18 19:22:23 -08:00
utils.py	Add a compile option to nn.value_and_grad	2024-01-21 21:25:01 -08:00