geohot 1037defce5 working WMMA

2025-08-24 16:46:48 -07:00

..

apps

BPE tokenizer (#11415)

2025-08-04 09:52:38 -07:00

codegen

working WMMA

2025-08-24 16:46:48 -07:00

engine

rangeify load cleanups + multi support (#11765)

2025-08-20 20:55:49 -07:00

frontend

onnx RotaryEmbedding cleanup (#11724)

2025-08-18 23:34:42 -04:00

nn

[bounty] Muon optim (#11414)

2025-08-13 14:27:55 -04:00

renderer

move into codegen late [pr] (#11823)

2025-08-24 10:23:25 -07:00

runtime

amd aql queue (#11708)

2025-08-24 19:53:00 +03:00

schedule

optional postopt

2025-08-24 14:42:31 -07:00

shape

add AxisType to range (#11798)

2025-08-23 11:15:00 -07:00

uop

Fast idiv try removing factors of two before cast (#11824)

2025-08-24 20:04:25 +02:00

viz

viz memory layout cleanup (#11820)

2025-08-24 19:37:31 +03:00

__init__.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

device.py

viz memory: compute nbytes (#11795)

2025-08-23 17:34:07 +03:00

dtype.py

generic double cast folding (#11481)

2025-08-02 19:26:37 -04:00

gradient.py

Revert "REDUCE_AXIS keepdim=False (#11311)" (#11718)

2025-08-18 13:28:53 -07:00

helpers.py

simpler postrange

2025-08-24 14:04:19 -07:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

Use Tensor.logaddexp to implement Tensor.softplus (#11796)

2025-08-23 11:52:29 -04:00