| .. |
|
codegen
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
engine
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
frontend
|
add onnx frontend stub [pr] (#9558)
|
2025-03-24 12:24:34 +08:00 |
|
nn
|
nn.state docs cleanup (#8332)
|
2025-03-18 17:16:40 -04:00 |
|
renderer
|
label ranges with their number (#9805)
|
2025-04-09 14:31:18 +08:00 |
|
runtime
|
amd: refactor sqtt into sep functions (#9816)
|
2025-04-10 00:39:45 +03:00 |
|
shape
|
cache folded upcast [pr] (#9733)
|
2025-04-04 11:23:19 +08:00 |
|
viz
|
add err handling tests to viz + cleanups (#9825)
|
2025-04-10 14:05:05 +08:00 |
|
__init__.py
|
UOp.variable (#7010)
|
2024-10-12 18:20:44 +08:00 |
|
device.py
|
Revert "FP8 support on NVIDIA (#8631)"
|
2025-04-09 12:27:41 +08:00 |
|
dtype.py
|
Revert "FP8 support on NVIDIA (#8631)"
|
2025-04-09 12:27:41 +08:00 |
|
gradient.py
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
helpers.py
|
use functools.cache instead of lru_cache(None) [pr] (#9714)
|
2025-04-03 11:47:13 +08:00 |
|
ops.py
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
py.typed
|
add a single py.typed (#6083)
|
2024-08-14 17:31:46 -07:00 |
|
spec.py
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
tensor.py
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
