George HotzandGitHub 2e7adb529f don't run kernels with 1000x more compute (fix BEAM with FUSE_ARANGE) (#5924)

2024-08-05 16:28:09 -07:00

..

codegen

fast mnist indexing (#5921)

2024-08-05 13:55:15 -07:00

engine

don't run kernels with 1000x more compute (fix BEAM with FUSE_ARANGE) (#5924)

2024-08-05 16:28:09 -07:00

nn

pow(2) -> square in RMSNorm [run_process_replay] (#5901)

2024-08-04 14:21:31 -04:00

renderer

hotfix: TC renders nv_bfloat16 (#5913)

2024-08-05 18:40:31 +03:00

runtime

metal sync queues in transfer (#5308)

2024-08-05 18:43:22 +03:00

shape

unify negative shape creation to raise ValueError (#5817)

2024-07-30 13:42:59 -04:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

device.py

hcq copy queue is optional (#5909)

2024-08-05 14:03:25 +03:00

dtype.py

support using str to specify dtype (#5897)

2024-08-04 12:56:28 -04:00

function.py

threefry as pattern matcher (#5371)

2024-07-13 11:59:03 -07:00

helpers.py

fast mnist indexing (#5921)

2024-08-05 13:55:15 -07:00

lazy.py

fast mnist indexing (#5921)

2024-08-05 13:55:15 -07:00

multi.py

MLB support reshape for uneven shards (#5804)

2024-08-01 02:36:03 -07:00

ops.py

add test to validate lazyops dims (#5845)

2024-07-31 12:59:38 -07:00

tensor.py

tracemeta fixups (#5904)

2024-08-04 16:15:06 -07:00