dm-haiku
JAX-based neural network library
AI summary
Neural Network Library
A JAX-based neural network library for building and optimizing neural networks
- stars
- 2.9K
- forks
- 232
- watching
- 37
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
google/flax6.2K
Neural Network Library
Provides a flexible neural network library and ecosystem for JAX
Neural Network Library
A simple, composable neural network library built on top of TensorFlow 2.
Graph library
A lightweight library for working with graph neural networks in jax.
JAX tutorial series
A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.
Research codebase
Provides implementations and illustrative code to accompany DeepMind research publications
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
SNN library
A JAX-based library for training and utilizing spiking neural networks
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
JAX helper library
A set of utilities for writing reliable JAX code
Deep learning library
An end-to-end deep learning library with clear code and speed
ResNet library
Provides implementations and checkpoints for various ResNet variants using JAX and Flax.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Deep neural network
An implementation of a deep neural network architecture for image classification using pre-trained models and fine-tuning on the CIFAR-10 dataset.
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Neural Network Library
A high-level neural network API for defining and training complex hierarchical networks of finite or infinite width
Featured in 1 awesome list
Each link jumps to the spot where the list mentions dm-haiku.