Skip to content

Commits

Branch selector

User selector

Datepicker

Commit History

Commits on Sep 26, 2025

[trax] Explicitly set `jax_pmap_shmap_merge=False`.
danielsuocopybara-github
danielsuo
authored and
copybara-github
committed

Commits on Aug 27, 2025

Remove more dtype canonicalization.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Apr 10, 2025

Configure tests to explicitly use jax_threefry_partitionable=False.
jburnimcopybara-github
jburnim
authored and
copybara-github
committed

Commits on Feb 7, 2025

Automated Code Change
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed

Commits on Feb 5, 2025

Avoid use of deprecated xla_bridge.get_backend().live_buffers()
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Jan 29, 2025

remove deprecated assertDictContainsSubset
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed

Commits on Jan 8, 2025

Future-proof calls to jnp.solve on batched 1D inputs.
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Sep 10, 2024

disable certain tests in lax_numpy_test when numpy 2.0 is used
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed
disable testOp when numpy 2.0 is installed
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed
fix a few test bitwise op tests for np 2.0
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed

Commits on Aug 21, 2024

skip testLinSpace and testLogSpace if numpy 2.0 is used
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed

Commits on Aug 18, 2024

[numpy] Fix test failures under NumPy 2.0.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Aug 13, 2024

[numpy] Fix test failures under NumPy 2.0.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Jul 16, 2024

[scipy] Use NumPy functions rather than their deprecated aliases in SciPy's namespace.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on May 16, 2024

Replace deprecated `jax.tree_*` functions with `jax.tree.*`
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Feb 1, 2024

Remove uses of deprecated jax.lax.tie_in
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Nov 15, 2023

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Oct 23, 2023

trax: avoid referencing deprecated jax.random.threefry_2x32
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Oct 20, 2023

Updated jax.config import
superbobrycopybara-github
superbobry
authored and
copybara-github
committed

Commits on Mar 29, 2023

Remove references to deprecated jax.ShapedArray
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Feb 15, 2023

[JAX] Replace uses of jax.xla_computation() with jax.jit().lower().
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Dec 19, 2022

[NumPy] Remove references to deprecated NumPy type aliases.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Nov 9, 2022

[tf.numpy] Fixes jax_tests/ breakages.
wangpengmitcopybara-github
wangpengmit
authored and
copybara-github
committed

Commits on Sep 9, 2022

Make this code compatible with Python 3.10.
yileicopybara-github
yilei
authored and
copybara-github
committed

Commits on Aug 8, 2022

[NumPy] Replace numpy.asscalar(x) with x.item() in preparation for upgrading NumPy to 1.23.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Aug 5, 2022

A change to type checking in checkpoint loading.
henrykmichalewskicopybara-github
henrykmichalewski
authored and
copybara-github
committed

Commits on Aug 4, 2022

[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.
hawkinspcopybara-github
hawkinsp
authored and
copybara-github
committed

Commits on Jun 27, 2022

Disable broken trax tests
Jake VanderPlascopybara-github
Jake VanderPlas
authored and
copybara-github
committed

Commits on Jun 7, 2022

Internal change
trax-robotcopybara-github
trax-robot
authored and
copybara-github
committed

Commits on May 17, 2022

Internal change
blakehechtmancopybara-github
blakehechtman
authored and
copybara-github
committed

Commits on Apr 24, 2022

Remove unused comments related to Python 2 compatibility.
yileicopybara-github
yilei
authored and
copybara-github
committed

Commits on Apr 22, 2022

Remove unused comments related to Python 2 compatibility.
yileicopybara-github
yilei
authored and
copybara-github
committed

Commits on Apr 21, 2022

Remove unused comments related to Python 2 compatibility.
yileicopybara-github
yilei
authored and
copybara-github
committed

Commits on Apr 20, 2022

Remove unused comments related to Python 2 compatibility.
yileicopybara-github
yilei
authored and
copybara-github
committed
Remove unused comments related to Python 2 compatibility.
yileicopybara-github
yilei
authored and
copybara-github
committed

Pagination