Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 26, 2022

• [JAX] Handle non-contiguous GPU IDs in NCCL collectives.

  Show description for b9d7e05
  

  hawkinsp

  authored and

  jax authors

  committed
  b9d7e05

  Copy full SHA for b9d7e05

• Merge pull request jax-ml#12127 from hawkinsp:ducc2

  Show description for 6205702
  

  jax authors

  committed
  6205702

  Copy full SHA for 6205702

• Add missing header to ducc build file.

  hawkinsp

  committed
  20475cc

  Copy full SHA for 20475cc

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

  Show description for f68f1c0
  

  hawkinsp

  authored and

  jax authors

  committed
  f68f1c0

  Copy full SHA for f68f1c0

• Merge pull request jax-ml#12125 from hawkinsp:xla

  Show description for 45764ea
  

  jax authors

  committed
  45764ea

  Copy full SHA for 45764ea

• Merge pull request jax-ml#12122 from hawkinsp:fft

  Show description for 498fd20
  

  jax authors

  committed
  498fd20

  Copy full SHA for 498fd20

• Bump XLA version.

  hawkinsp

  committed
  12cb55d

  Copy full SHA for 12cb55d

• Migrate from MLIR-HLO's CHLO to StableHLO's CHLO

  Show description for 2186268
  

  Eugene Burmako

  authored and

  jax authors

  committed
  2186268

  Copy full SHA for 2186268

• Merge pull request jax-ml#12101 from sharadmv:perfetto-link

  Show description for 1329591
  

  jax authors

  committed
  1329591

  Copy full SHA for 1329591

• Fixes for PocketFFT->ducc migration.

  Show description for b63801b
  hawkinsp

  committed
  b63801b

  Copy full SHA for b63801b

• Switch from pocketfft to ducc

  Show description for 024ae47
  

  Edenhofer

  authored and

  hawkinsp

  committed
  024ae47

  Copy full SHA for 024ae47

• Merge pull request jax-ml#12121 from hawkinsp:iree

  Show description for 4aa96c0
  

  jax authors

  committed
  4aa96c0

  Copy full SHA for 4aa96c0

• Update IREE backend name from "dylib" to "llvm-cpu".

  hawkinsp

  committed
  813ee0d

  Copy full SHA for 813ee0d

• [jax2tf] Improves support for lax.gather (enable_xla=False).

  Show description for 6dd425c
  

  marcvanzee

  authored and

  jax authors

  committed
  6dd425c

  Copy full SHA for 6dd425c

• Add runtime type check in named_scope to ensure that name is a string.

  Show description for 3c91787
  

  levskaya

  authored and

  jax authors

  committed
  3c91787

  Copy full SHA for 3c91787

• Merge pull request jax-ml#12113 from sudhakarsingh27:combine_gpu_tests_rebased

  Show description for 0cbdf85
  

  jax authors

  committed
  0cbdf85

  Copy full SHA for 0cbdf85

• Add support for MeshPspecSharding local_sharded_result_handler because SDA outputs from pjit can produce a MeshPspecSharding.

  Show description for 96058d0
  

  yashk2810

  authored and

  jax authors

  committed
  96058d0

  Copy full SHA for 96058d0

Commits on Aug 25, 2022

• combine gpu tests

  sudhakarsingh27

  committed
  4b1a2ea

  Copy full SHA for 4b1a2ea

• Copybara import of the project:

  Show description for cc8d406
  

  jax authors

  committed
  cc8d406

  Copy full SHA for cc8d406

• Add option to generate perfetto trace without generating link

  sharadmv

  committed
  260f1d8

  Copy full SHA for 260f1d8

• Merge pull request jax-ml#12091 from jakevdp:getitem-slice

  Show description for 1a36534
  

  jax authors

  committed
  1a36534

  Copy full SHA for 1a36534

• generate lax.slice instead of lax.gather for more indexing cases

  jakevdp

  committed
  d5ee729

  Copy full SHA for d5ee729

• Merge pull request jax-ml#12107 from jakevdp:fix-pmap-test

  Show description for 7219cdc
  

  jax authors

  committed
  7219cdc

  Copy full SHA for 7219cdc

• pmap_test: disable DeviceBufferToArray when jax_array=True

  jakevdp

  committed
  e98b955

  Copy full SHA for e98b955

• minimal result-handling support for single-device key array pjit outputs

  Show description for acc025a
  

  2 peopleauthored and

  jax authors

  committed
  acc025a

  Copy full SHA for acc025a

• Annotate TypeAlias

  Show description for 0ff0417
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  0ff0417

  Copy full SHA for 0ff0417

• Fix jax2tf GDA bug.

  Show description for 28b14b1
  

  maxwillzq

  authored and

  jax authors

  committed
  28b14b1

  Copy full SHA for 28b14b1

• Introduce class PyShardedBuffer that contains a vector of PjRtBuffer and cache it in GDA.

  Show description for 12f85c8
  

  cky9301

  authored and

  jax authors

  committed
  12f85c8

  Copy full SHA for 12f85c8

• Merge pull request jax-ml#12081 from jakevdp:fix-unary-test

  Show description for a5a7445
  

  jax authors

  committed
  a5a7445

  Copy full SHA for a5a7445

• Merge pull request jax-ml#12088 from jakevdp:bcoo-slice

  Show description for 06bd056
  

  jax authors

  committed
  06bd056

  Copy full SHA for 06bd056

• Merge pull request jax-ml#12037 from luca-digrazia:patch-1

  Show description for d605eca
  

  jax authors

  committed
  d605eca

  Copy full SHA for d605eca

• [sparse] add support for bcoo equivalent of lax.slice

  jakevdp

  committed
  269e752

  Copy full SHA for 269e752

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

  Show description for 5527966
  

  hawkinsp

  authored and

  jax authors

  committed
  5527966

  Copy full SHA for 5527966

• __repr__ if an Array is fully replicated. Its the same for _value so it makes sense to do the same for __repr__.

  Show description for fd3a72d
  

  yashk2810

  authored and

  jax authors

  committed
  fd3a72d

  Copy full SHA for fd3a72d

• support jax.experimental.array.Array as a base array for key arrays

  Show description for 8e2d1be
  

  2 peopleauthored and

  jax authors

  committed
  8e2d1be

  Copy full SHA for 8e2d1be

Pagination

Previous

Next