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