Skip to content

nbxtensor 0.0.4

Choose a tag to compare

@benkelaya benkelaya released this 07 Nov 00:52
· 118 commits to main since this release

Added

  • Triton for the Apple GPU in one command: python -m nbxtensor install-metal installs the Metal wheels of the
    project's fork NeuroBrix/triton-ext (Triton built with triton-ext, and imperatormk's AppleGPU backend as
    nbx-triton-apple, triton-lang/triton-ext#126), each by the release URL and sha256 written in the library,
    through pip with --no-deps --require-hashes --no-index — never an extra index nor a package name resolved on PyPI. The
    wheels are for Python 3.12 and macOS 15 or later (release apple-v0.1.0+triton3.8.0); it refuses by name off an
    Apple-silicon Mac, on another Python, or below the macOS the wheels are built for. python -m nbxtensor doctor on a Mac
    without it says so and names the command; the Metal backend's refusal names it too. On a Mac, nbxtensor depends on
    pyobjc-framework-Metal, through which it opens the Metal device; without it the run fell to a CPU profile and was
    refused for triton-cpu, it is now refused by name with its install line.

Fixed

  • nbxtensor installs on a Mac: it depended on triton on every platform, and PyPI publishes no Triton for macOS, so
    pip install nbxtensor — and with it the NeuroBrix engine, even its PyTorch engine on MPS — could not resolve on a
    clean macOS environment. Triton is now a dependency where PyPI has it; on a Mac it is built from source with its Metal
    backend (the README's install), and the library says by name what to build when it is absent.