• 3
• 0
• 0

jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

• 3
• 0
• 0

flows_for_atomic_solids

• 3
• 1
• 0

ithaca

Restoring and attributing ancient texts using deep neural networks

• 3
• 0
• 0

distribution_shift_framework

This repository contains the code of the distribution shift framework presented in A Fine-Grained Analysis on Distribution Shift (Wiles et al., 2022).

• 3
• 0
• 0

local_linearity_regularizer

• 3
• 0
• 0

code_contests

• 3
• 0
• 0

affordances_option_models

• 3
• 0
• 0

neural_lns

• 3
• 0
• 0

dm_hamiltonian_dynamics_suite

• 3
• 0
• 0

dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.

• 3
• 0
• 0

brave

A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.

• 3
• 0
• 0

neural_testbed

• 3
• 0
• 0

robotics

Libraries, tools and tasks created and used at DeepMind Robotics.

• 3
• 0
• 0

alphafold

Open source code for AlphaFold 2.

• 2
• 0
• 0

educational

• 3
• 0
• 0

einshape

• 2
• 0
• 0

jmp

JMP is a Mixed Precision library for JAX.

• 2
• 0
• 0

tensor_annotations

Annotating tensor shapes using Python types

• 2
• 0
• 0

chex

• 2
• 0
• 0

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations