trellis-2-mrp-mlx/trellis2/modules/sparse
2026-07-17 09:27:07 +03:00
..
attention sparse/attention: drop FUSED_ATTN_MAX_SEQLEN threshold to match CUDA-path precedent 2026-07-17 07:51:05 +03:00
conv add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
spatial Init Release 2025-12-16 19:00:42 +00:00
transformer Init Release 2025-12-16 19:00:42 +00:00
__init__.py Init Release 2025-12-16 19:00:42 +00:00
basic.py Init Release 2025-12-16 19:00:42 +00:00
config.py test: probe production split-k Metal kernel 2026-07-17 09:27:07 +03:00
linear.py Init Release 2025-12-16 19:00:42 +00:00
nonlinearity.py Init Release 2025-12-16 19:00:42 +00:00
norm.py sparse/norm: CPU-staged zeros_like workaround for MPS 2026-07-17 07:51:05 +03:00