PIX is an image processing library in JAX, for JAX.

Python389apache-2.0

5 days ago

computer-visionimageimage-processing

Python98apache-2.0

5 months ago

A customisable 3D platform for agent-based AI research

C7133other

2 years ago

artificial-intelligencedeep-learningmachine-learning

Monte Carlo tree search in JAX

Python2353apache-2.0

4 months ago

jaxmonte-carlo-tree-searchreinforcement-learning

Question answering dataset featured in "Teaching Machines to Read and Comprehend

Python1293apache-2.0

8 years ago

Python105apache-2.0

9 days ago

Python788apache-2.0

9 hours ago

JAX-based neural network library

Python2904apache-2.0

6 days ago

deep-learningdeep-neural-networksjax

A Graph Neural Network Library in Jax

Python1374apache-2.0

8 months ago

deep-learninggraph-neural-networksjax

Python1261apache-2.0

2 months ago

Open source code for AlphaFold 2.

Python12814apache-2.0

5 months ago

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Python248apache-2.0

yesterday

bayesian-deep-learningmachine-learningoptimization

This repository contains the NarrativeQA dataset. It includes the list of docume

Shell457apache-2.0

5 years ago

Cephes Mathematical Functions library wrapped for Torch

C47other

9 years ago

Provides and wraps the Randomkit library, copied from Numpy.

Lua34bsd-3-clause

6 years ago

Optax is a gradient processing and optimization library for JAX.

Python1693apache-2.0

yesterday

machine-learningoptimization