70cd436eb9
Extracted from lato.2_mrp_mlx. The same sparse module underlies LATO.2, Pixal3D and the rest of the TRELLIS.2 family, and all of them are blocked on Apple Silicon by the same single op - submanifold conv - so it belongs in one tested package rather than vendored per port. 13/13 tests: 7 for the conv against a hand-written reference (spconv is uninstallable here so there is no upstream oracle), 6 for the remaining layers against torch. SubMConv3d runs 13.8ms at 128^3/128ch on m3ultra, 5.9x faster than the obvious per-offset loop.
15 lines
402 B
TOML
15 lines
402 B
TOML
[project]
|
|
name = "trellis-sparse-mlx"
|
|
version = "0.1.0"
|
|
description = "MLX implementation of the TRELLIS-lineage sparse module (SubMConv3d et al) for Apple Silicon"
|
|
requires-python = ">=3.10"
|
|
dependencies = ["mlx>=0.20", "numpy>=1.24"]
|
|
license = { text = "MIT" }
|
|
|
|
[build-system]
|
|
requires = ["setuptools>=68"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[tool.setuptools]
|
|
packages = ["trellis_sparse_mlx"]
|