Skip to content

NeuroBrix 0.6.4

Choose a tag to compare

@benkelaya benkelaya released this 09 Nov 17:08
· 101 commits to main since this release

NeuroBrix 0.6.4

NeuroBrix 0.6.4 is a corrective release of 0.6.3, with nbxtensor 0.0.9.

  • macOS 14 and CPython 3.10–3.13 on Apple GPUs. The macOS 14 Metal wheels (Python 3.10–3.13) are published (triton-ext release); python -m nbxtensor install-metal now selects them: on macOS 14 every covered Python gets the macOS 14 build, and Python 3.12 on macOS 15 or later keeps the macOS 15 build. The Triton engine runs on macOS 14: its CPU pool no longer needs Accelerate's thread API of macOS 15, and python -m nbxtensor doctor says when Accelerate's threads are not controlled (VECLIB_MAXIMUM_THREADS=1, set before Python starts, is the user's choice).
  • A first launch with the Triton engine no longer stops when the reload it times holds a zero-size constant (Sana 1024's text encoder on a fresh machine with 0.6.3).

Known limit: no macOS 14 machine has run this release: the evidence is Accelerate's macOS 14 BLAS simulated on a newer macOS (TinyLlama, whisper and Sana outputs equal by sha) and all 238 Metal kernels of those three models compiled for macOS 14's Metal language (Metal 3.1, -mmacosx-version-min=14.0) without an error.

Install: pip install neurobrix (the Triton engine, no PyTorch) or pip install "neurobrix[pytorch]"; on a Mac, python -m nbxtensor install-metal. The library is nbxtensor 0.0.9.

The full list is in the CHANGELOG.