Awni Hannun 63ae767232 fix transformer (#1327)		2024-08-13 16:04:26 -07:00
..
layers	fix transformer (#1327)	2024-08-13 16:04:26 -07:00
__init__.py	Common neural network initializers nn.initializers (#456)	2024-01-23 06:47:20 -08:00
init.py	post nanobind docs fixes and some updates (#889)	2024-03-24 15:03:27 -07:00
losses.py	Improve stability of BCE loss calculation for input probabilities close to or exactly 0 or 1 (#1280)	2024-07-24 08:38:22 -07:00
utils.py	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00