Flax is a neural network library for JAX
A JAX research toolkit to build, edit, & visualize neural networks
TensorFlow-based neural network library
A Graph Neural Network Library in Jax
A library for easy and efficient manipulation of tensor networks
This is the code for Deformable Neural Radiance Fields
A model library for exploring state-of-the-art deep learning
Time Series Forecasting Best Practices & Examples
Basic Utilities for PyTorch Natural Language Processing (NLP)
Convolutional Neural Network for Text Classification in Tensorflow