nnx
Dia-JAX: A JAX port of Dia, the text-to-speech model for generating realistic dialogue from text with emotion and tone control.
🐦⬛🐦⬛A high-level API to build and train Flax/NNX models at scale
Differentiable audio for JAX, written in Faust and compiled to Flax NNX
