monster/trellis-2-mrp-mlx
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
7860148eb2
trellis-2-mrp-mlx/mlx_backend
History
m3ultra 7860148eb2
Some checks are pending
CodeQL Advanced / Analyze (${{ matrix.language }}) (none, c-cpp) (push) Waiting to run
Details
CodeQL Advanced / Analyze (${{ matrix.language }}) (none, python) (push) Waiting to run
Details
spconv Metal kernel (opt-in) + fix vertex baker scheduler fallthrough 
- sparse_conv_metal: fused gather-GEMM kernel, parity 8e-4 vs stock;
  honest verdict: stock chunking already bounds memory at decoder scale
  (1.97GB synthetic peak) and beats the scalar kernel on speed — kept as
  TRELLIS2_METAL_SPCONV=1 opt-in + upstream reference, NOT default.
- _attempt_schedule: preferred=vertex now runs the vertex baker (was
  silently falling through to the 5h+ pure-python kdtree grind).

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-20 03:37:20 +10:00
..
__init__.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
adapters.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
attention.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
convert.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
dinov3.py feat: add native Apple Silicon asset pipeline 2026-07-17 09:21:34 +03:00
flow_models.py inline env defaults in api_server, remove start script, drop intermediate mx.eval syncs 2026-07-17 07:51:05 +03:00
mlx_samplers.py mlx_backend: pure-MLX FlowEuler+CFG(+interval) sampler loop 2026-07-20 03:01:57 +10:00
norm.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
pipeline.py mlx_backend: pure-MLX FlowEuler+CFG(+interval) sampler loop 2026-07-20 03:01:57 +10:00
rope.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
sparse_conv_metal.py spconv Metal kernel (opt-in) + fix vertex baker scheduler fallthrough 2026-07-20 03:37:20 +10:00
sparse_conv.py spconv Metal kernel (opt-in) + fix vertex baker scheduler fallthrough 2026-07-20 03:37:20 +10:00
sparse_ops.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
sparse_tensor.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
structure_decoder.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
transformer_block.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
vae_blocks.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00
vae_decoders.py add mlx backend for apple silicon with metal gpu acceleration 2026-07-17 07:51:05 +03:00