Reviews
Optax is a gradient processing and optimization library for JAX.
Search similar apps
License
Apache License 2.0
Optax is a gradient processing and optimization library for JAX.
Creator
Related apps
dm_pix
PIX is an image processing library in JAX, for JAX.
Python389apache-2.0
5 days ago
computer-visionimageimage-processing
lab
A customisable 3D platform for agent-based AI research
C7133other
2 years ago
artificial-intelligencedeep-learningmachine-learning
mctx
Monte Carlo tree search in JAX
Python2353apache-2.0
4 months ago
jaxmonte-carlo-tree-searchreinforcement-learning
rc-data
Question answering dataset featured in "Teaching Machines to Read and Comprehend
Python1293apache-2.0
8 years ago
dm-haiku
JAX-based neural network library
Python2904apache-2.0
6 days ago
deep-learningdeep-neural-networksjax
jraph
A Graph Neural Network Library in Jax
Python1374apache-2.0
8 months ago
deep-learninggraph-neural-networksjax
kfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Python248apache-2.0
yesterday
bayesian-deep-learningmachine-learningoptimization
narrativeqa
This repository contains the NarrativeQA dataset. It includes the list of docume
Shell457apache-2.0
5 years ago
torch-randomkit
Provides and wraps the Randomkit library, copied from Numpy.
Lua34bsd-3-clause
6 years ago