Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 27, 2022

• Add Github Actions workflow that runs on a self-hosted TPU VM runner.

  Show description for 1c572d3
  skye

  committed
  1c572d3

  Copy full SHA for 1c572d3

Commits on Oct 21, 2022

• First pass at adding `GetOutputShardings` and `GetParameterShardings` on PjRTExecutable.

  Show description for 37a0156
  

  yashk2810

  authored and

  jax authors

  committed
  37a0156

  Copy full SHA for 37a0156

• Merge pull request #12899 from jakevdp:dead-code

  Show description for e297b13
  

  jax authors

  committed
  e297b13

  Copy full SHA for e297b13

• Merge pull request #12920 from yilei:nameerror

  Show description for 045d6e1
  

  jax authors

  committed
  045d6e1

  Copy full SHA for 045d6e1

• Flatten the if/else block.

  yilei

  committed
  d63d038

  Copy full SHA for d63d038

• Fix an NameError caused by #12754

  yilei

  authored
  7c1bf0e

  Copy full SHA for 7c1bf0e

• Merge pull request #12904 from froystig:dlpack64

  Show description for 503679d
  

  jax authors

  committed
  503679d

  Copy full SHA for 503679d

• [MLIR] Don't rely on hardcoded -1 for dynamic axis sizes

  Show description for dd04953
  

  d0k

  authored and

  jax authors

  committed
  dd04953

  Copy full SHA for dd04953

• Merge pull request #12876 from mattjj:djax-vmap3

  Show description for 2228efe
  

  jax authors

  committed
  2228efe

  Copy full SHA for 2228efe

• fix

  mattjj

  committed
  8e8ae84

  Copy full SHA for 8e8ae84

• put back some unsafe_maps

  mattjj

  committed
  f76fc01

  Copy full SHA for f76fc01

• Merge pull request #12912 from mattjj:issue12909

  Show description for 8030b49
  

  jax authors

  committed
  8030b49

  Copy full SHA for 8030b49

• update docs to point to jax.nn.standardize

  Show description for 6a3d2a0
  mattjj

  committed
  6a3d2a0

  Copy full SHA for 6a3d2a0

• Merge pull request #12800 from mattjj:ayaka

  Show description for 2d45d8b
  

  jax authors

  committed
  2d45d8b

  Copy full SHA for 2d45d8b

• improve (and shorten!) pmap error messages about inconsistent axis sizes

  mattjj

  committed
  60b236c

  Copy full SHA for 60b236c

Commits on Oct 20, 2022

• Merge pull request #12906 from jakevdp:stats-mode-doc

  Show description for 1b5294e
  

  jax authors

  committed
  1b5294e

  Copy full SHA for 1b5294e

• fix jax2tf readme typo

  Show description for 408953b
  

  maxwillzq

  authored and

  jax authors

  committed
  408953b

  Copy full SHA for 408953b

• Merge pull request #12896 from jakevdp:unused-imports

  Show description for 6e4b135
  

  jax authors

  committed
  6e4b135

  Copy full SHA for 6e4b135

• Add docs & changelog for jax.scipy.stats.mode

  jakevdp

  committed
  4aceb81

  Copy full SHA for 4aceb81

• canonicalize dtypes when loading arrays via `dlpack`

  froystig

  committed
  4af3509

  Copy full SHA for 4af3509

• Cleanup: remove unused imports in private modules

  Show description for 7f89fd4
  jakevdp

  committed
  7f89fd4

  Copy full SHA for 7f89fd4

• Merge pull request #12872 from jakevdp:annotate-map

  Show description for 2801533
  

  jax authors

  committed
  2801533

  Copy full SHA for 2801533

• Merge pull request #12900 from jakevdp:dlpack-test

  Show description for f7f2351
  

  jax authors

  committed
  f7f2351

  Copy full SHA for f7f2351

• Test: skip JaxToNumy DLPack test on numpy 1.22

  Show description for dcd9143
  jakevdp

  committed
  dcd9143

  Copy full SHA for dcd9143

• Remove unused code from jax/_src/lax/linalg.py

  jakevdp

  committed
  d6d5339

  Copy full SHA for d6d5339

• Merge pull request #12894 from jakevdp:annotate-ufuncs

  Show description for 4facacb
  

  jax authors

  committed
  4facacb

  Copy full SHA for 4facacb

• [sparse] Update the default cuSparse matvec algorithm in jaxlib.

  Show description for 7093142
  

  tlu7

  authored and

  jax authors

  committed
  7093142

  Copy full SHA for 7093142

• [typing] annotate jax.numpy ufuncs

  jakevdp

  committed
  6d30865

  Copy full SHA for 6d30865

• [sparse] Add BCSR primitive bcsr_extract.

  Show description for 2d563bf
  

  tlu7

  authored and

  jax authors

  committed
  2d563bf

  Copy full SHA for 2d563bf

• [typing] annotate jax._src.util.safe_map

  jakevdp

  committed
  5d15757

  Copy full SHA for 5d15757

• Move `is_deleted()` to C++ so that we can check if an Array is deleted without materializing `_arrays`.

  Show description for 1a0affd
  

  yashk2810

  authored and

  jax authors

  committed
  1a0affd

  Copy full SHA for 1a0affd

• Merge pull request #12798 from skye:cache_min_instr_count

  Show description for c1c8462
  

  jax authors

  committed
  c1c8462

  Copy full SHA for c1c8462

• Add new config `jax_persistent_cache_min_instruction_count`.

  Show description for 81eb3fc
  skye

  committed
  81eb3fc

  Copy full SHA for 81eb3fc

• [dynamic-shapes] fix nested vmap callable annotation logic

  mattjj

  committed
  a1d303b

  Copy full SHA for a1d303b

Commits on Oct 19, 2022

• Merge pull request #12871 from ROCmSoftwarePlatform:rocm_docker_fix_for_pip

  Show description for 8536fae
  

  jax authors

  committed
  8536fae

  Copy full SHA for 8536fae

Pagination

Previous

Next