George Hotz
0cbf6c1811
move things, clean up extra ( #2292 )
...
* move things
* idk why pylint needs that now
* delete unused
2023-11-13 20:18:40 -08:00
geohotstan
8d6cecb25c
Torch eq fix ( #1562 )
...
* init
* Revert "init"
This reverts commit 682bf2073a
2023-10-11 12:57:11 -07:00
George Hotz
a6d842af7a
move device to ops ( #1646 )
...
* move device to ops
* mlops types
* 2 lines
2023-08-23 08:30:17 -07:00
George Hotz
718ced296c
move state to nn/state ( #1619 )
2023-08-22 07:36:24 -07:00
JaSpa99
d3d58a37e5
Bert: use Tensor.scaled_dot_product_attention ( #1528 )
...
* use scaled attn from Tensor
* add a test for bert
* linter
* no more tokenizer
* without loading weights
* remove prints
* tribute to linter lords
* smaller input and less runs
* small bert
2023-08-12 08:46:04 -07:00
