MORPH
SIGN IN SIGN UP
jax-ml / jax UNCLAIMED

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

35249 0 0 Python

ACTIVITY

Commits Reviews
week month year
COMMITS
20
in the last week
CONTRIBUTORS
10
active
STARS
35249
total
FORKS
0
total
TOP CONTRIBUTORS
J
jax authors
5 commits
J
Jake VanderPlas
4 commits
Y
Yash Katariya
3 commits
J
Jacob Burnim
2 commits
R
Robert Dyro
1 commit
P
Peter Hawkins
1 commit
V
Vladimir Belitskiy
1 commit
S
Sergei Lebedev
1 commit
N
Nick Groszewski
1 commit
T
Tomás Longeri
1 commit
RECENT COMMITS
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/dd42b1de151120245edaf199313d66a48fe20e7a
jax authors
84df746
J
Use logging instead of absl.logging in interpret/shared_memory.py
Jacob Burnim
a96db3f
Y
Only log elapsed time during tracing for top level jit
Yash Katariya
cea418a
R
Rename `manual_type=.../.manual_type/.mt` to `manual_axis_type=.../.manual_axis_type/.mat` to avoid confusion with `jnp.array(...).mT`
Robert Dyro
6735ea0
J
Merge pull request #36302 from jakevdp:del-get-aval
jax authors
8799b33
J
Merge pull request #36292 from jakevdp:argsort-stable
jax authors
a1747e1
P
Make CPU devices report as cpu:0 instead of TFRT_CPU_0.
Peter Hawkins
5be30dd
J
Remove deprecated symbol `jax.interpreters.batching.NotMapped`.
Jake VanderPlas
6e1c296
J
Remove unused logging import in pjit.py
Jacob Burnim
8de54a1
J
Remove internal get_aval alias
Jake VanderPlas
1c31181
V
Ignore the missing attr error for `Jaxpr._debug_info` for now.
Vladimir Belitskiy
5f26b11
S
Removed a few unused classes
Sergei Lebedev
edaec9a
J
[test] remove obsolete NumPy compatibility code
Jake VanderPlas
f050845
J
Merge pull request #36189 from groszewn:num_eqns
jax authors
163aa54
N
Track total number of eqns contained in a jaxpr
Nick Groszewski
8fec761
J
Merge pull request #36093 from jpbrodrick89:jpb/scipy-batching
jax authors
8842cf8
Y
Make is_equivalent_to work with AbstractMesh
Yash Katariya
3f1d12f
Y
Fix top_k's sharding rule to error out if the input array is sharded over the axis along which top_k values are computed because that will lead to comms.
Yash Katariya
521fa8b
T
[Mosaic] Better type constraints for conversion ops
Tomás Longeri
06d5f6c
J
Remove deprecated jax.core.[un]mapped_aval for JAX v0.10.0.
Jake VanderPlas
db61be4