Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 17, 2024

• Reverts 525b646

  Show description for 7de9eb2
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  7de9eb2

  Copy full SHA for 7de9eb2

Commits on Dec 16, 2024

• Fix some flaky LAX autodiff tests.

  Show description for a2e9699
  

  dfm

  authored and

  Google-ML-Automation

  committed
  a2e9699

  Copy full SHA for a2e9699

Commits on Dec 13, 2024

• Use a broadcasted gather in the sort JVP, rather than forming explicit iotas.

  Show description for 0922feb
  hawkinsp

  committed
  0922feb

  Copy full SHA for 0922feb

Commits on Dec 12, 2024

• Disable failing test cases when JAX_ENABLE_X64=1 in the Bazel CPU build

  Show description for ecc2673
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  ecc2673

  Copy full SHA for ecc2673

Commits on Nov 19, 2024

• Reverts 2075b09

  Show description for 525b646
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  525b646

  Copy full SHA for 525b646

• Add a new API jax.lax.split.

  Show description for 2c80d1a
  hawkinsp

  committed
  2c80d1a

  Copy full SHA for 2c80d1a

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

  Copy full SHA for d4d1518

Commits on Apr 11, 2024

• Avoid 'from jax import config' imports

  Show description for f090074
  jakevdp

  committed
  f090074

  Copy full SHA for f090074

Commits on Dec 22, 2023

• del add_any_p and zeros_like_p, replace aval-dispatched traceable

  mattjj

  committed
  be3ca50

  Copy full SHA for be3ca50

Commits on Oct 12, 2023

• MAINT Migrate remaining internal/test modules to use state objects

  Show description for cbcaac2
  superbobry

  committed
  cbcaac2

  Copy full SHA for cbcaac2

Commits on Oct 7, 2023

• fix pow jvp rule with int exponent (broken since jax-ml#16419)

  Show description for 6fe5731
  mattjj

  committed
  6fe5731

  Copy full SHA for 6fe5731

Commits on Sep 27, 2023

• Add a new internal test utility test_device_matches() and use it instead of equality tests on device_under_test().

  Show description for 1885c49
  

  hawkinsp

  authored and

  jax authors

  committed
  1885c49

  Copy full SHA for 1885c49

Commits on Aug 25, 2023

• lax.pow: fix shape mismatch failure in jvp rule

  jakevdp

  committed
  6cec5d4

  Copy full SHA for 6cec5d4

Commits on Aug 23, 2023

• Merge pull request jax-ml#16419 from mattjj:pow-jvp

  Show description for af42359
  

  jax authors

  committed
  af42359

  Copy full SHA for af42359

Commits on Aug 7, 2023

• ENH: Update numpy exceptions imports

  mtsokol

  committed
  d183a2c

  Copy full SHA for d183a2c

Commits on Jul 29, 2023

• fix pow_p jvp rule at x=0. y=0

  Show description for 69ad4df
  

  mattjj

  and

  froystig

  committed
  69ad4df

  Copy full SHA for 69ad4df

Commits on Jul 28, 2023

• add an exp2 primitive and lax.exp2

  Show description for 560ede0
  mattjj

  committed
  560ede0

  Copy full SHA for 560ede0

Commits on Jul 13, 2023

• make scatter-apply jvp notimplemented (for now...)

  Show description for 2cfffb6
  

  3 peoplecommitted
  2cfffb6

  Copy full SHA for 2cfffb6

Commits on Jun 23, 2023

• Relax test tolerances.

  Show description for 0adfafe
  

  hawkinsp

  authored and

  jax authors

  committed
  0adfafe

  Copy full SHA for 0adfafe

Commits on May 22, 2023

• allow lax.dot_general to accept different input dtypes

  Show description for 61b106e
  mattjj

  committed
  61b106e

  Copy full SHA for 61b106e

Commits on Apr 21, 2023

• Change to simpler import for jax.config

  jakevdp

  committed
  fbe4f10

  Copy full SHA for fbe4f10

Commits on Feb 28, 2023

• Replace jax._src.util.prod with math.prod.

  Show description for 8fb1fd3
  

  hawkinsp

  authored and

  jax authors

  committed
  8fb1fd3

  Copy full SHA for 8fb1fd3

Commits on Dec 21, 2022

• relax tanh test tols for upcoming xla change

  mattjj

  committed
  6ba0ef6

  Copy full SHA for 6ba0ef6

Commits on Dec 14, 2022

• adjust tolerance for dot_general gradient test on TPU

  Show description for a66b3dc
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  a66b3dc

  Copy full SHA for a66b3dc

Commits on Nov 10, 2022

• Rename test flag --num_generated_cases to --jax_num_generated_cases.

  Show description for e42e52d
  

  hawkinsp

  authored and

  jax authors

  committed
  e42e52d

  Copy full SHA for e42e52d

Commits on Oct 14, 2022

• Fix lax_autodiff_test to avoid scatters into overlapping ranges.

  Show description for 5d07cbe
  

  hawkinsp

  authored and

  jax authors

  committed
  5d07cbe

  Copy full SHA for 5d07cbe

Commits on Oct 12, 2022

• Migrate remaining tests from jtu.cases_from_list to jtu.sample_product.

  Show description for 72f4f38
  hawkinsp

  committed
  72f4f38

  Copy full SHA for 72f4f38

• Reenable some tests disabled in the past because of an LLVM bug.

  Show description for 9bb2c99
  

  hawkinsp

  authored and

  jax authors

  committed
  9bb2c99

  Copy full SHA for 9bb2c99

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

  Copy full SHA for ba557d5

Commits on Sep 7, 2022

• Roll forward: Upgrade logistic into a primitive.

  Show description for 57b5acf
  

  hawkinsp

  authored and

  jax authors

  committed
  57b5acf

  Copy full SHA for 57b5acf

Commits on Aug 30, 2022

• Rollback of upgrade logistic (sigmoid) function into a lax primitive.

  Show description for 9c16c83
  

  jax authors

  committed
  9c16c83

  Copy full SHA for 9c16c83

Commits on Aug 26, 2022

• Upgrade logistic (sigmoid) function into a lax primitive.

  Show description for f68f1c0
  

  hawkinsp

  authored and

  jax authors

  committed
  f68f1c0

  Copy full SHA for f68f1c0

Commits on Aug 24, 2022

• Upgrade logistic (sigmoid) function into a lax primitive.

  Show description for 3e3542b
  

  jax authors

  committed
  3e3542b

  Copy full SHA for 3e3542b

• Upgrade logistic (sigmoid) function into a lax primitive.

  Show description for 6276194
  

  hawkinsp

  authored and

  jax authors

  committed
  6276194

  Copy full SHA for 6276194

Commits on Aug 23, 2022

• Fix JVP rule for lax.pow()

  jakevdp

  committed
  b8fe0ab

  Copy full SHA for b8fe0ab

Pagination

Previous

Next