Commits

Branch selector

 main

User selector

shoyer

Datepicker

All time

Commit History

Commits on May 3, 2024

• Recursively pull out __wrapped__ in linkcode_resolve()

  Show description for c77370c
  shoyer

  committed
  c77370c

  Copy full SHA for c77370c

• Make linkcode_resolve() a bit more robust

  Show description for f6ee612
  shoyer

  committed
  f6ee612

  Copy full SHA for f6ee612

Commits on May 12, 2023

• Fix type annotations on jax.vmap

  shoyer

  committed
  3254bf8

  Copy full SHA for 3254bf8

Commits on Apr 3, 2023

• Support extrapolation in jnp.interp

  Show description for 4009005
  shoyer

  committed
  4009005

  Copy full SHA for 4009005

Commits on Mar 7, 2023

• Add "compare_all" method to searchsorted

  shoyer

  committed
  d4f70c8

  Copy full SHA for d4f70c8

Commits on Jul 14, 2022

• Fix stacklevel for _check_user_dtype_supported

  Show description for 36c601c
  shoyer

  committed
  36c601c

  Copy full SHA for 36c601c

Commits on Oct 29, 2021

• Generate a better error message if [pv]map receives no array arguments

  Show description for 860e5e7
  shoyer

  committed
  860e5e7

  Copy full SHA for 860e5e7

Commits on Sep 2, 2021

• Don't refer to deprecated jax.ops.index_update in error messages

  Show description for d204325
  shoyer

  committed
  d204325

  Copy full SHA for d204325

Commits on Aug 30, 2021

• Add jit to lax.fft

  Show description for 22943ef
  shoyer

  committed
  22943ef

  Copy full SHA for 22943ef

• New implementation of jnp.polyval using lax.scan

  shoyer

  committed
  9df340d

  Copy full SHA for 9df340d

Commits on Jun 2, 2021

• Merge branch 'master' into master

  shoyer

  authored
  5b3d203

  Copy full SHA for 5b3d203

Commits on Jun 1, 2021

• Fix wrong method name

  shoyer

  authored
  d098391

  Copy full SHA for d098391

Commits on Apr 7, 2021

• add fallback for inspect.signature inside jit

  shoyer

  committed
  ebc86f8

  Copy full SHA for ebc86f8

Commits on Mar 31, 2021

• Add _python_jit_with_static_argnames.

  shoyer

  committed
  acb0be9

  Copy full SHA for acb0be9

Commits on Mar 26, 2021

• Fix transpose rule for jnp.fft.irfft

  Show description for bc2e428
  shoyer

  committed
  bc2e428

  Copy full SHA for bc2e428

Commits on Mar 1, 2021

• Fix data placement reference

  shoyer

  committed
  ac9b1ce

  Copy full SHA for ac9b1ce

Commits on Feb 28, 2021

• DOC: Add a FAQ on "Benchmarking JAX code"

  Show description for 46db7bb
  shoyer

  committed
  46db7bb

  Copy full SHA for 46db7bb

Commits on Feb 10, 2021

• Better error messages for test_util.check_grads()

  Show description for db6405c
  shoyer

  committed
  db6405c

  Copy full SHA for db6405c

Commits on Dec 4, 2020

• Support ndarrays as arguments to cg and gmres

  Show description for cd9f6cc
  shoyer

  committed
  cd9f6cc

  Copy full SHA for cd9f6cc

Commits on Dec 3, 2020

• Cleanup/fixup jax.scipy.sparse.linalg.gmres and expose it publicly.

  shoyer

  committed
  6cc5b28

  Copy full SHA for 6cc5b28

Commits on Nov 12, 2020

• Relax GMRES test tolerances

  shoyer

  committed
  fca5666

  Copy full SHA for fca5666

Commits on Nov 11, 2020

• More unit-tests + mark gmres as internal for now

  shoyer

  committed
  7e62270

  Copy full SHA for 7e62270

Commits on Nov 10, 2020

• Add a FAQ entry on slow JIT compilation

  shoyer

  committed
  bb684b5

  Copy full SHA for bb684b5

Commits on Nov 9, 2020

• Cleanup _safe_normalize

  shoyer

  committed
  c5c71a0

  Copy full SHA for c5c71a0

Commits on Oct 29, 2020

• Add reference on RK45

  shoyer

  committed
  3c5f82f

  Copy full SHA for 3c5f82f

Commits on Oct 15, 2020

• Support specifying precision as a tuple of precision enums

  Show description for 706bab7
  shoyer

  committed
  706bab7

  Copy full SHA for 706bab7

Commits on Sep 17, 2020

• Add jax.linear_transpose (jax-ml#3398)

  Show description for 877053d
  

  shoyer

  and

  mattjj

  authored
  877053d

  Copy full SHA for 877053d

Commits on Sep 14, 2020

• Simplify the interface for host_callback.id_tap (jax-ml#4101)

  Show description for 6bd3216
  shoyer

  authored
  6bd3216

  Copy full SHA for 6bd3216

Commits on Aug 20, 2020

• Document the required form of tap_func for host_callback.id_tap (jax-ml#4100)

  shoyer

  authored
  d978808

  Copy full SHA for d978808

Commits on Aug 18, 2020

• Add experimental __array_module__ method (jax-ml#4076)

  Show description for decd760
  shoyer

  authored
  decd760

  Copy full SHA for decd760

Commits on Aug 6, 2020

• Fix jax.checkpoint in API docs (jax-ml#3980)

  Show description for 816bcd7
  shoyer

  authored
  816bcd7

  Copy full SHA for 816bcd7

Commits on Jul 29, 2020

• Fixes to test_scipy_optimize.py for Google internal tests (jax-ml#3902)

  shoyer

  authored
  b0ef483

  Copy full SHA for b0ef483

• Add missing license headers (jax-ml#3899)

  Show description for 242b324
  shoyer

  authored
  242b324

  Copy full SHA for 242b324

Commits on Jul 28, 2020

• Fix formatting in the custom derivatives notebook (jax-ml#3876)

  Show description for dd7ab39
  shoyer

  authored
  dd7ab39

  Copy full SHA for dd7ab39

Commits on Jul 24, 2020

• Create a separate internal helper function for XLA compilation (jax-ml#3852)

  Show description for b7bcfa6
  shoyer

  authored
  b7bcfa6

  Copy full SHA for b7bcfa6

Pagination

Previous

Next