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 |
---|---|---|
| Provides a flexible neural network library and ecosystem for JAX | 6,196 |
| A simple, composable neural network library built on top of TensorFlow 2. | 9,790 |
| A lightweight library for working with graph neural networks in jax. | 1,380 |
| A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. | 670 |
| Provides implementations and illustrative code to accompany DeepMind research publications | 13,329 |
| Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
| A JAX-based library for training and utilizing spiking neural networks | 104 |
| Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
| A set of utilities for writing reliable JAX code | 797 |
| An end-to-end deep learning library with clear code and speed | 8,114 |
| Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 105 |
| Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
| An implementation of a deep neural network architecture for image classification using pre-trained models and fine-tuning on the CIFAR-10 dataset. | 285 |
| An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
| A high-level neural network API for defining and training complex hierarchical networks of finite or infinite width | 2,291 |