Commit Graph

1 Commits

Author SHA1 Message Date
George Hotz 643cbdfd50
make embedding and GPT-2 fast (#1631)
* make embedding fast

* jit more, variable shape support

* print mem bw
2023-08-22 15:14:38 -07:00