Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 12, 2024

• Don't monkey-patch functions in test_utils to count events for tests.

  Show description for 62e66b6
  hawkinsp

  committed
  62e66b6

Commits on Sep 20, 2024

• Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

  Show description for d4d1518
  

  MichaelHudgins

  authored and

  Google-ML-Automation

  committed
  d4d1518

Commits on Jul 15, 2024

• [ROCM] test_computation_follows_data fix for rocm and cuda

  Ruturaj4

  committed
  fb5c516

Commits on Jun 13, 2024

• fix some additional warnings related to jax-ml#21834

  jakevdp

  committed
  3346527

Commits on Jun 6, 2024

• test cleanup: use ExitStack to reduce test boilerplate

  jakevdp

  committed
  a861c55

Commits on Apr 11, 2024

• Avoid 'from jax import config' imports

  Show description for f090074
  jakevdp

  committed
  f090074

Commits on Feb 13, 2024

• Simplifies full to not instantiate intermediate array with default sharding, this significantly reduces overhead when creating sharded arrays in eager mode when using jnp.zeros_like(...)

  Show description for 2717dae
  

  marksandler2

  authored and

  jax authors

  committed
  2717dae

Commits on Jan 23, 2024

• Test: move lax.full sharding tests to multi_device_test

  jakevdp

  committed
  8476b44

Commits on Nov 30, 2023

• Deprecate the device() method of JAX arrays

  jakevdp

  committed
  97beb01

• fix grad device_put src inference, and a small device_put bug

  Show description for c9ab0bf
  

  mattjj

  and

  yashk2810

  committed
  c9ab0bf

Commits on Aug 28, 2023

• Tests: set jax_legacy_prng_key='error'

  jakevdp

  committed
  2f878a7

Commits on Aug 25, 2023

• Remove trivial execution from jax since it leads to 100x slower dispatch time.

  Show description for 970f4c9
  

  yashk2810

  authored and

  jax authors

  committed
  970f4c9

Commits on Apr 21, 2023

• Change to simpler import for jax.config

  jakevdp

  committed
  fbe4f10

Commits on Mar 16, 2023

• Remove references to jax.config.jax_jit_pjit_api_merge, which is always True at head.

  Show description for 1813553
  

  yashk2810

  authored and

  jax authors

  committed
  1813553

• Remove references to jax.config.jax_array, which is always True at head.

  Show description for dea7450
  

  hawkinsp

  authored and

  jax authors

  committed
  dea7450

Commits on Feb 28, 2023

• [JAX] Move jax._src.lib.xla_bridge to jax._src.xla_bridge.

  Show description for f66f6ec
  

  hawkinsp

  authored and

  jax authors

  committed
  f66f6ec

Commits on Feb 10, 2023

• Improve the error message which is raised from _get_and_check_device_assignment.

  Show description for 1526c3e
  

  yashk2810

  authored and

  jax authors

  committed
  1526c3e

Commits on Jan 9, 2023

• Fix the error message of different devices when jit/pjit are merged

  Show description for 74601e5
  

  yashk2810

  authored and

  jax authors

  committed
  74601e5

Commits on Sep 26, 2022

• Rename the concrete class Array to ArrayImpl

  Show description for cbf34cb
  

  yashk2810

  authored and

  jax authors

  committed
  cbf34cb

Commits on Sep 22, 2022

• Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

  Show description for ba557d5
  

  hawkinsp

  authored and

  jax authors

  committed
  ba557d5

Commits on Sep 9, 2022

• Minimally support device argument on jit in the jax.Array path

  Show description for 980aa31
  

  yashk2810

  authored and

  jax authors

  committed
  980aa31

Commits on Sep 8, 2022

• Merge pull request jax-ml#12297 from mattjj:computation-follows-data-prng

  Show description for edfbbd7
  

  jax authors

  committed
  edfbbd7

• add _device attribute to PRNGKeyArray so that computation follows key placement

  Show description for 47b2dfe
  mattjj

  committed
  47b2dfe

• Fix Forward. The fix is on the user's end. Original PR: jax-ml#12217

  Show description for 7fbf8ec
  

  2 peopleauthored and

  jax authors

  committed
  7fbf8ec

• roll back breakage

  Show description for 14f1a34
  

  jax authors

  committed
  14f1a34

Commits on Sep 6, 2022

• DCE jaxpr and trivial_jaxpr support for lower_sharding_computation

  Show description for b7e4e44
  

  2 peopleauthored and

  jax authors

  committed
  b7e4e44

Commits on Aug 30, 2022

• Make jit == pjit. This means that the lowering and execution paths of jit and pjit are merged.

  Show description for 6340952
  

  yashk2810

  authored and

  jax authors

  committed
  6340952

Commits on Aug 19, 2022

• Use .device() to get the device and platform from the device and fix TODO to point to github issue

  Show description for 314cf8a
  

  yashk2810

  authored and

  jax authors

  committed
  314cf8a

• Enable jax_array on CPU for the entire JAX test suite!

  Show description for d77848b
  

  yashk2810

  authored and

  jax authors

  committed
  d77848b

Commits on Mar 1, 2022

• Update JAX to use new math libraries in ROCm-5.0.

  reza-amd

  committed
  a0d9d81

Commits on Nov 22, 2021

• Split jax.interpreters.xla up into three pieces:

  Show description for d262bae
  

  hawkinsp

  authored and

  jax authors

  committed
  d262bae

Commits on Sep 24, 2021

• Move jax.test_util to jax._src.test_util.

  Show description for db2e91e
  

  hawkinsp

  authored and

  jax authors

  committed
  db2e91e

Commits on Sep 23, 2021

• Move contents of jax.lib to jax._src.lib.

  Show description for 2c2f403
  

  hawkinsp

  authored and

  jax authors

  committed
  2c2f403

Commits on Mar 29, 2021

• remove disable_omnistaging mechanism

  mattjj

  committed
  2b79264

Commits on Mar 10, 2021

• Remove the JAX lazy sublanguage.

  Show description for 140c0ac
  hawkinsp

  committed
  140c0ac

Pagination

Previous

Next