dm_pix

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

License

Apache License 2.0

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

Creator

deepmind

Related apps

distrax

Python446apache-2.0

8 months ago

einshape

einshape

Python77apache-2.0

last year

lab

lab

A customisable 3D platform for agent-based AI research

C6891other

last year

artificial-intelligencedeep-learningmachine-learning

mctx

Monte Carlo tree search in JAX

Python1946apache-2.0

8 months ago

jaxmonte-carlo-tree-searchreinforcement-learning

rc-data

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

Python1283apache-2.0

7 years ago

tf2jax

Python81apache-2.0

8 months ago

trfl

TensorFlow Reinforcement Learning

Python3142apache-2.0

last year

chex

Python580apache-2.0

10 months ago

dm-haiku

JAX-based neural network library

Python2546apache-2.0

9 months ago

deep-learningdeep-neural-networksjax

jraph

jraph

A Graph Neural Network Library in Jax

Python1184apache-2.0

last year

deep-learninggraph-neural-networksjax

rlax

Python1055apache-2.0

10 months ago

sonnet

sonnet

TensorFlow-based neural network library

Python9586apache-2.0

10 months ago

artificial-intelligencedeep-learningmachine-learning

alphafold

alphafold

Open source code for AlphaFold.

Python10533apache-2.0

10 months ago

kfac-jax

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

Python153apache-2.0

8 months ago

bayesian-deep-learningmachine-learningoptimization

narrativeqa

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

Shell414apache-2.0

4 years ago

torch-cephes

Cephes Mathematical Functions library wrapped for Torch

C46other

8 years ago

torch-randomkit

Provides and wraps the Randomkit library, copied from Numpy.

Lua34bsd-3-clause

5 years ago

optax

Optax is a gradient processing and optimization library for JAX.

Python1144apache-2.0

10 months ago