Justin Deschenaux 8e5600022a Implement RNN, GRU, LSTM (#268) ... * RNN base implementation * Address comments+format * nits in docs * add tests for prb * fix test * add a couple tests --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-03-11 21:14:44 -07:00
..
layers	Implement RNN, GRU, LSTM (#268)	2024-03-11 21:14:44 -07:00
__init__.py	Common neural network initializers nn.initializers (#456)	2024-01-23 06:47:20 -08:00
init.py	updated calls to use loc &scale (#643)	2024-02-08 09:01:59 -08:00
losses.py	Feat: Add weights argument in BCE Loss and tests (#620)	2024-02-07 09:39:52 -08:00
utils.py	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00