chex
AI summary
JAX helper library
A set of utilities for writing reliable JAX code
- stars
- 797
- forks
- 44
- watching
- 17
- awesome lists
- 2
Similar projects
Found by comparing what the projects do, not just their names.
Graph library
A lightweight library for working with graph neural networks in jax.
Reshaper
A unified reshaping library for JAX and other frameworks.
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
Gradient optimizer library
A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components.
Data navigator
A library for working with nested data structures in Python
Collage generator
A toolkit for generating collage images using neural networks and optimization algorithms
Search tool
Tools for performing searches against a Google custom search engine and returning unique page results
Android environment
A platform for defining reinforcement learning tasks on top of Android devices
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Meta learning experiment
A repository containing code for a meta-learning experiment on image datasets
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions chex.
