jraph

A Graph Neural Network Library in Jax

License

Apache License 2.0

A Graph Neural Network Library in Jax

Creator

google-deepmind

Related apps

distrax

Python537apache-2.0

2 months ago

dm_pix

dm_pix

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

Python390apache-2.0

15 days ago

computer-visionimageimage-processing

einshape

einshape

Python99apache-2.0

5 months ago

lab

lab

A customisable 3D platform for agent-based AI research

C7134other

2 years ago

artificial-intelligencedeep-learningmachine-learning

mctx

Monte Carlo tree search in JAX

Python2359apache-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

tf2jax

Python105apache-2.0

19 days ago

chex

Python788apache-2.0

2 days ago

dm-haiku

JAX-based neural network library

Python2909apache-2.0

15 days ago

deep-learningdeep-neural-networksjax

rlax

Python1264apache-2.0

2 months ago

alphafold

alphafold

Open source code for AlphaFold 2.

Python12892apache-2.0

5 days ago

kfac-jax

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

Python249apache-2.0

yesterday

bayesian-deep-learningmachine-learningoptimization

narrativeqa

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

Shell459apache-2.0

5 years ago

torch-cephes

Cephes Mathematical Functions library wrapped for Torch

C47other

9 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.

Python1705apache-2.0

11 days ago

machine-learningoptimization