-
vq-vae Public
A JAX / NNX implementation of a VQ-VAE for audio compression
Jupyter Notebook UpdatedNov 8, 2024 -
nnx-gru Public
A JAX / NNX implementation of a linear GRU for language modeling
Jupyter Notebook UpdatedOct 31, 2024 -
biggan Public
BigGAN in idiomatic Keras
-
pytorch Public
Forked from pytorch/pytorchTensors and Dynamic neural networks in Python with strong GPU acceleration
C++ Other UpdatedMar 8, 2021 -
tensorflow Public
Forked from tensorflow/tensorflowAn Open Source Machine Learning Framework for Everyone
C++ Apache License 2.0 UpdatedApr 21, 2019