Reviews
JAX-based neural network library
Search similar apps
License
Apache License 2.0
Related apps
dm_pix
PIX is an image processing library in JAX, for JAX.
Python389apache-2.0
4 days ago
computer-visionimageimage-processing
lab
A customisable 3D platform for agent-based AI research
C7130other
2 years ago
artificial-intelligencedeep-learningmachine-learning
mctx
Monte Carlo tree search in JAX
Python2351apache-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
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
21 hours ago
bayesian-deep-learningmachine-learningoptimization
narrativeqa
This repository contains the NarrativeQA dataset. It includes the list of docume
Shell458apache-2.0
5 years ago
torch-randomkit
Provides and wraps the Randomkit library, copied from Numpy.
Lua34bsd-3-clause
6 years ago
optax
Optax is a gradient processing and optimization library for JAX.
Python1692apache-2.0
yesterday
machine-learningoptimization