Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 15, 2023

• migrate more internal dependencies from jax.core to jax._src.core

  Show description for cb8dcce
  

  froystig

  authored and

  jax authors

  committed
  cb8dcce

• Add clear_cache endpoint to python pjit and cpp pjit functions.

  Show description for b476661
  

  yashk2810

  authored and

  jax authors

  committed
  b476661

Commits on Dec 27, 2022

• Error on numpy masked array inputs.

  jakevdp

  committed
  5367693

Commits on Dec 19, 2022

• Bump the minimum jaxlib version to 0.4.1.

  Show description for 2c6c30d
  hawkinsp

  committed
  2c6c30d

Commits on Nov 15, 2022

• Disable some tests with jax.Array that are failing in OSS due to using minimum_jaxlib_version. I will bump the version again this week.

  Show description for eca1241
  

  yashk2810

  authored and

  jax authors

  committed
  eca1241

Commits on Sep 26, 2022

• Change jax_jit_test to be a jax_test() under Bazel that works across backends.

  Show description for d63a944
  

  hawkinsp

  authored and

  jax authors

  committed
  d63a944

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 Aug 25, 2022

• [JAX] Deprecate .to_py() property on arrays. Implement __array__ instead.

  Show description for 5527966
  

  hawkinsp

  authored and

  jax authors

  committed
  5527966

Commits on Aug 23, 2022

• [JAX] Prepare not to export jax._src by default.

  Show description for 335b2cf
  

  hawkinsp

  authored and

  jax authors

  committed
  335b2cf

Commits on May 18, 2022

• feat: validate jit args

  JeppeKlitgaard

  committed
  838a053

Commits on Jan 24, 2022

• jax_jit_test: inherit from JaxTestCase

  jakevdp

  committed
  8cacad2

Commits on Dec 7, 2021

• Move dtype canonicalization out of core.AbstractValue subclasses.

  Show description for 06cd1fe
  

  hawkinsp

  authored and

  jax authors

  committed
  06cd1fe

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 Jun 2, 2021

• Move jax.abstract_arrays to jax._src.abstract_arrays.

  Show description for 46cc654
  

  hawkinsp

  authored and

  jax authors

  committed
  46cc654

Commits on Apr 13, 2021

• Move jax.api to jax._src.api.

  Show description for 26e9ebc
  

  hawkinsp

  authored and

  jax authors

  committed
  26e9ebc

Commits on Mar 16, 2021

• Increase minimum jaxlib version to 0.1.62.

  hawkinsp

  committed
  328930b

Commits on Mar 7, 2021

• [XLA:Python] Split DevicePut out of jax_jit and refactor it.

  Show description for 2722837
  

  hawkinsp

  authored and

  jax authors

  committed
  2722837

Commits on Feb 4, 2021

• Cleanup: remove obsolete jaxlib version checks

  jakevdp

  committed
  5e7be4a

• Make jax_enable_x64 a thread-local value.

  jakevdp

  committed
  2fd682e

Commits on Feb 2, 2021

• Fix a number of NumPy 1.20.0 test failures.

  hawkinsp

  committed
  5fa48aa

Commits on Jan 14, 2021

• Disable a JAX jit test that fails with the released Jaxlib 0.1.58 (which appears to have xla_extension version number 2). Bump the XLA client version number.

  Show description for b1f1e31
  

  hawkinsp

  authored and

  jax authors

  committed
  b1f1e31

Commits on Jan 12, 2021

• Add a function to get the signature associated to an argument.

  Show description for 92c993a
  

  jblespiau

  authored and

  jax authors

  committed
  92c993a

• Add support for more numpy types, and use a map-lookup design.

  Show description for 30f73f8
  

  jblespiau

  authored and

  jax authors

  committed
  30f73f8

Commits on Jan 11, 2021

• Add workaround for pybind11 bug that means we were issue poor errors in the case of Python integer -> C integer conversion overflow.

  Show description for 2ec9b13
  

  hawkinsp

  authored and

  jax authors

  committed
  2ec9b13

Commits on Oct 14, 2020

• Fix typo.

  jblespiau

  committed
  47319b8

Commits on Oct 12, 2020

• Copybara import of the project:

  Show description for ee9ca56
  

  jblespiau

  authored and

  jax authors

  committed
  ee9ca56

• Add support for jax_debug_nans and fix the last few glitches with the C++ jax.jit.

  Show description for c1e2595
  

  jblespiau

  authored and

  jax authors

  committed
  c1e2595

Commits on Oct 8, 2020

• Fallback to Python for float0 in the C++ jax.jit.

  Show description for 008eae6
  

  jblespiau

  authored and

  jax authors

  committed
  008eae6

• remove stale version guard in test

  froystig

  committed
  1fb097b

• skip test that is broken at unreleased (source) jaxlib

  froystig

  committed
  16cd330

Commits on Oct 7, 2020

• Ignore an incorrect mypy error.

  jblespiau

  committed
  ab7436a

Commits on Sep 1, 2020

• Add more features to the C++ jax.jit. (jax-ml#4169)

  Show description for bdd6545
  jblespiau

  authored
  bdd6545

Commits on Aug 18, 2020

• Use pytree defined in tensorflow. (jax-ml#4087)

  Show description for 2ab6b42
  jblespiau

  authored
  2ab6b42

Loading