dm-haiku
Neural Network Library
A JAX-based neural network library for building and optimizing neural networks
JAX-based neural network library
3k stars
37 watching
231 forks
Language: Python
last commit: 15 days ago
Linked from 1 awesome list
deep-learningdeep-neural-networksjaxmachine-learningneural-networks
Related projects:
| Repository | Description | Stars |
|---|---|---|
 google/flax | A high-performance neural network library for JAX designed to facilitate flexibility and ease of use. | 6,146 |
| google-deepmind/sonnet | A simple, composable neural network library built on top of TensorFlow 2. | 9,782 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
 gordicaleksa/get-started-with-jax | A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. | 661 |
| google-deepmind/deepmind-research | Provides implementations and illustrative code to accompany DeepMind research publications | 13,274 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 kmheckel/spyx | A JAX-based library for training and utilizing spiking neural networks | 101 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 788 |
| google/trax | An end-to-end deep learning library with clear code and speed | 8,102 |
 n2cholas/jax-resnet | Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 104 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |
 conan7882/googlenet-inception | An implementation of a deep neural network architecture for image classification using pre-trained models and fine-tuning on the CIFAR-10 dataset. | 282 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 389 |
| google/neural-tangents | A high-level neural network API for defining and training complex hierarchical networks of finite or infinite width | 2,278 |