boris-kuz/snax

jax port of snac

stars

commits

Python

primary language

May 12, 2024

updated

README

* SNAX

Port of [[https://github.com/hubertsiuzdak][Hubert Siuzdak's]] [[https://github.com/hubertsiuzdak/snac][snac]] to [[https://github.com/google/jax][JAX]] & [[https://github.com/patrick-kidger/equinox][Equinox]].

On my system with a RTX 2080, I get a ~100x speedup when transcoding a 9s long stereo snippet after jit-ing.
More thorough benchmarks to be conducted.

Not written in Markdown, so it's shown here as plain text — view it formatted on GitHub.

Contributors

boris-kuz

5 commits

Similar repos

sanchit-gandhi/whisper-jax

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

4,681

yangwhale/gpu-tpu-pedia

pasttofuture-whisperer/xprof-jitter-interceptor

Deterministic flow control architecture to eliminate JAX/XLA dynamic recompilation and physical…

felafax/HiFi-GAN-TPU

Running FastPitch + HifiGAN model on TPUs using Torch:XLA

huckiyang/jax-nemotron

open hymba nemotron on jax / nnx

patil-suraj/vqgan-jax

JAX implementation of VQGAN

hanjq17/Spectrum

[CVPR 2026] Adaptive Spectral Feature Forecasting for Diffusion Sampling Acceleration

133

mlnomadpy/gemma4-jax

From-scratch Flax NNX port of google/gemma-4-12B — full encoder-free multimodal (text + vision +…

Languages

Python

100.0%

README

* SNAX

Port of [[https://github.com/hubertsiuzdak][Hubert Siuzdak's]] [[https://github.com/hubertsiuzdak/snac][snac]] to [[https://github.com/google/jax][JAX]] & [[https://github.com/patrick-kidger/equinox][Equinox]].

On my system with a RTX 2080, I get a ~100x speedup when transcoding a 9s long stereo snippet after jit-ing.
More thorough benchmarks to be conducted.

Not written in Markdown, so it's shown here as plain text — view it formatted on GitHub.