George Hotz f494b9d463 simple multitensor API (#2903) ...

* simple multitensor API

* test multitensor

* mt work

* new api

* copies

* all but data parallel

* allreduce there

* works, but axis sharded

* fix all mt tests

* features/multi

* work

* backprop

* fix tests

* tests passing

* mt progress

* cleanups

* less lines

* tensor cleanup

* save more lines

* mypy passes

* fix tests

* skip for cuda too

* bump download cache

2024-01-02 17:49:44 -08:00

..

benchmark.yml

switch CI to tiny8 (#2984)

2024-01-02 16:40:25 -08:00

python-publish.yml

allow manual release (#1704)

2023-08-28 17:54:25 -07:00

szdiff.yml

keep only one comment from git action bot (#1936)

2023-09-28 20:24:53 -04:00

test.yml

simple multitensor API (#2903)

2024-01-02 17:49:44 -08:00