Audio data loading and augmentations in JAX
MIT License
A JAX Implementation of the Descript Audio Codec
Implementation of https://srush.github.io/annotated-s4
Versatile audio super resolution (any -> 48kHz) with AudioSR.
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...
Turn SymPy expressions into trainable JAX expressions.
AudioLDM: Generate speech, sound effects, music and beyond, with text.
A Jax wrapper for cudaKDTree
Pytrees + dataclasses ❤️
Attax: adversarial attacks using JAX
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Frechet Inception Distance in JAX
A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Multiple dispatch over abstract array types in JAX.