chex

Search similar apps

    License

    Apache License 2.0

    Creator

    google-deepmind

    Related apps

    distrax

    Python536apache-2.0

    2 months ago

    dm_pix

    dm_pix

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

    Python389apache-2.0

    5 days ago

    computer-visionimageimage-processing

    einshape

    einshape

    Python98apache-2.0

    5 months ago

    lab

    lab

    A customisable 3D platform for agent-based AI research

    C7133other

    2 years ago

    artificial-intelligencedeep-learningmachine-learning

    mctx

    Monte Carlo tree search in JAX

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

    9 days ago

    dm-haiku

    JAX-based neural network library

    Python2904apache-2.0

    6 days ago

    deep-learningdeep-neural-networksjax

    jraph

    jraph

    A Graph Neural Network Library in Jax

    Python1374apache-2.0

    8 months ago

    deep-learninggraph-neural-networksjax

    rlax

    Python1261apache-2.0

    2 months ago

    alphafold

    alphafold

    Open source code for AlphaFold 2.

    Python12814apache-2.0

    5 months ago

    kfac-jax

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

    Python248apache-2.0

    yesterday

    bayesian-deep-learningmachine-learningoptimization

    narrativeqa

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

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

    Python1693apache-2.0

    yesterday

    machine-learningoptimization