3 people f520323054 optimizer: add matvec optimizations (#1753) ...

* optimizer: add matvec optimizations

* Update optimizer.py

---------

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

2023-10-03 00:01:59 -07:00

..

codegen

optimizer: add matvec optimizations (#1753)

2023-10-03 00:01:59 -07:00

nn

Refactor Conv2d/ConvTranspose2d into a single parent class (#1906)

2023-09-24 14:23:41 +08:00

renderer

fix llvm nan/inf const (#1951)

2023-10-02 03:08:57 -07:00

runtime

metal batch executor (#1920)

2023-10-02 03:18:31 -07:00

shape

perf: assert behind if is not optimized (#1847)

2023-09-29 11:07:24 -07:00

graph.py

more lazy cleanup (#1938)

2023-09-29 00:53:29 -07:00

helpers.py

Do not import typing_extensions at runtime (#1927)

2023-09-28 01:57:28 -07:00

jit.py

fix jitted dist (#1955)

2023-10-02 11:45:13 -04:00

lazy.py

fix llvm nan/inf const (#1951)

2023-10-02 03:08:57 -07:00

mlops.py

reorder a few things (#1915)

2023-09-25 10:17:21 +08:00

ops.py

metal batch executor (#1920)

2023-10-02 03:18:31 -07:00

tensor.py

simplify line (#1950)

2023-10-02 03:19:15 -07:00