Skip to content

Add mlx-sci (scientific computing toolkit)#5

Open
akaiHuang wants to merge 1 commit into
raullenchai:mainfrom
akaiHuang:add-mlx-sci
Open

Add mlx-sci (scientific computing toolkit)#5
akaiHuang wants to merge 1 commit into
raullenchai:mainfrom
akaiHuang:add-mlx-sci

Conversation

@akaiHuang

Copy link
Copy Markdown

Adds mlx-sci under Other Tools.

mlx-sci is a meta-package of nine focused MLX sub-packages providing GPU-accelerated scientific computing on Apple Silicon:

Special functions (mlx_sci.special):

  • Spherical Bessel j_l(x) — 579× over scipy at 525 ℓ × 10k pts
  • Wigner 3j/6j/9j and Clebsch-Gordan — 2000× batched over sympy
  • Gauss hypergeometric ₂F₁ with fused Metal kernel — 7.4× over scipy at N=1M
  • Airy Ai/Bi, gamma/lgamma/digamma

Linear algebra (mlx_sci.linalg): matrix expm / logm / sqrtm / Fréchet derivatives.

Signal processing (mlx_sci.signal): class-based STFT/ISTFT layers with mx.compile fusion — 10× over scipy on 30s audio.

Quantum information (mlx_sci.quantum):

  • Quantum relative entropy via Stochastic Lanczos Quadrature (650× over NumPy eigh at N=2000; only path that completes past N=2000 on Metal)
  • Petz recovery map, fidelity, retrodiction quality
  • Quantum channels (thermal attenuator, depolarising, dephasing)
  • Fisher information matrix, natural gradient

Install: pip install mlx-sci (PyPI, MIT, Python ≥3.10, MLX ≥0.30).

Each sub-package is independently pip-installable (mlx-bessel, mlx-qre, etc.) for à la carte usage. All headline numbers come with break-even tables and accuracy disclosures in each sub-package's benchmark_results.md.

mlx-sci ships GPU-accelerated scientific computing on Apple Silicon:
special functions (Bessel/Airy/gamma/hyp2f1/Wigner symbols), matrix
functions (expm/logm/sqrtm), STFT, and quantum-information primitives
including a Stochastic Lanczos Quadrature estimator for quantum
relative entropy and Petz recovery utilities.

PyPI: https://pypi.org/project/mlx-sci/
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant