This repository contains implementations and illustrative code to accompany DeepMind publications
Stars
14.8k
Forks
2.9k
Watchers
14.8k
Open Issues
335
Overall repository health assessment
No package.json found
This might not be a Node.js project
29
commits
22
commits
18
commits
16
commits
14
commits
12
commits
11
commits
9
commits
9
commits
8
commits
Replace references to deprecated jax.curry function.
9d01171View on GitHubFix pytype failures related to teaching pytype about NumPy scalar types.
82a3474View on GitHubReplaces references to jax.numpy.DeviceArray with jax.Array.
d988ff1View on GitHubSuppress some pytype errors related to jnp.DeviceArray == jax.Array.
784f675View on GitHubFix or ignore some pytype errors related to jnp.ndarray == jax.Array.
6f0ddefView on GitHubFix or ignore some pytype errors related to jnp.ndarray == jax.Array.
c051e6aView on GitHub[JAX] Remove obsolete unit type declarations in jax.core.
797ea3cView on GitHubUpdate DM21 pinned requirements and bazel config for compiling functional to C++.
cb555c2View on GitHub[NumPy] Remove references to deprecated NumPy type aliases.
c7e2ef2View on GitHub