Loreon
Labs
Platform
Docs
Home
Ecosystems
Other
distrax
8bitmp3/distrax
Other
Emerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
10mo ago
Recent commits
Latest commits.
remove `pjit_p`, leaving only `jax.extend.core.primitives.jit_p`
1f7ef15
Roy Frostig
11mo ago
[distrax] Replace usages of deprecated `jax.interpreters.xla.abstractify`
7fc5bd7
Jake VanderPlas
13mo ago
Configure tests to explicitly use jax_threefry_partitionable=False.
1e50d98
Jacob Burnim
14mo ago
Configure tests to explicitly use jax_threefry_partitionable=False.
efeeee8
Jacob Burnim
14mo ago
Add accuracy field to unary ops
c902a5b
Rachel Han
15mo ago
Fix wrong call signatures for jnp.shape, jnp.size, and jnp.ndim
8ffec28
Jake VanderPlas
16mo ago
Migrate from jax.core to jax.extend.core for several deprecated symbols
c02708a
Jake VanderPlas
18mo ago
Remove references to jax.core.raise_to_shaped
ddf4c7e
Jake VanderPlas
18mo ago
Top contributors
Builders behind this project.
franrruiz
50 commits
gpapamak
35 commits
hbq1
30 commits
kashif
23 commits
hawkinsp
10 commits
hamzamerzic
3 commits
jburnim
2 commits
mtthss
2 commits