Python446apache-2.0

8 months ago

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

Python316apache-2.0

10 months ago

computer-visionimageimage-processing

Python77apache-2.0

last year

A customisable 3D platform for agent-based AI research

C6891other

last year

artificial-intelligencedeep-learningmachine-learning

Monte Carlo tree search in JAX

Python1946apache-2.0

8 months ago

jaxmonte-carlo-tree-searchreinforcement-learning

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

Python1283apache-2.0

7 years ago

Python81apache-2.0

8 months ago

TensorFlow Reinforcement Learning

Python3142apache-2.0

last year

Python580apache-2.0

10 months ago

A Graph Neural Network Library in Jax

Python1184apache-2.0

last year

deep-learninggraph-neural-networksjax

Python1055apache-2.0

10 months ago

TensorFlow-based neural network library

Python9586apache-2.0

10 months ago

artificial-intelligencedeep-learningmachine-learning

Open source code for AlphaFold.

Python10533apache-2.0

10 months ago

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

Python153apache-2.0

8 months ago

bayesian-deep-learningmachine-learningoptimization

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

Shell414apache-2.0

4 years ago

Cephes Mathematical Functions library wrapped for Torch

C46other

8 years ago

Provides and wraps the Randomkit library, copied from Numpy.

Lua34bsd-3-clause

5 years ago

Optax is a gradient processing and optimization library for JAX.

Python1144apache-2.0

10 months ago