George Hotz 64563abc90 add LSTMCell to nn (#6080) ...

* add LSTMCell to nn

* lstmcell works with no input on first

* fix no bias 0

* simpler

2024-08-14 12:08:42 -07:00

..

__init__.py

add LSTMCell to nn (#6080)

2024-08-14 12:08:42 -07:00

datasets.py

hotfix: import datasets in nn init

2024-06-10 11:33:50 +02:00

optim.py

nn docs (#4735)

2024-05-26 13:08:22 -07:00

state.py

add Tensor.clamp and fix bool loading (#6069)

2024-08-13 15:26:40 -07:00