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
232 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
deep-learningdeep-neural-networksjaxmachine-learningneural-networks
Related projects:
| Repository | Description | Stars |
|---|---|---|
 google/flax | Provides a flexible neural network library and ecosystem for JAX | 6,196 |
| google-deepmind/sonnet | A simple, composable neural network library built on top of TensorFlow 2. | 9,790 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
 gordicaleksa/get-started-with-jax | A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. | 670 |
| google-deepmind/deepmind-research | Provides implementations and illustrative code to accompany DeepMind research publications | 13,329 |
 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 | 104 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 797 |
| google/trax | An end-to-end deep learning library with clear code and speed | 8,114 |
 n2cholas/jax-resnet | Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 105 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
 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. | 285 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
| google/neural-tangents | A high-level neural network API for defining and training complex hierarchical networks of finite or infinite width | 2,291 |