3 people a72b1b6d65 sharding for llama (#3151) ...

* shard llama

* sharding works

* simpler

* simpler

* consume option

* disable that test

* save a line

---------

Co-authored-by: George Hotz <[email protected]>

2024-01-16 19:28:00 -08:00

..

image.py

wmma: enable METAL half tensor cores and clean up cstyle (#3095)

2024-01-12 16:25:28 -05:00

multi.py

sharding for llama (#3151)

2024-01-16 19:28:00 -08:00

search.py

multitensor jit (#3149)

2024-01-16 09:09:15 -08:00