Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 24, 2024

• Update XLA dependency to use revision

  Show description for 0e5c44a
  

  jax authors

  committed
  0e5c44a

  Copy full SHA for 0e5c44a

Commits on Mar 23, 2024

• Adds additional check for x is a Tracer, as

  Show description for 1f81477
  

  marksandler2

  authored and

  jax authors

  committed
  1f81477

  Copy full SHA for 1f81477

• Update XLA dependency to use revision

  Show description for 30674ae
  

  jax authors

  committed
  30674ae

  Copy full SHA for 30674ae

• Update the arg description.

  Show description for b709925
  

  changhuilin

  authored and

  jax authors

  committed
  b709925

  Copy full SHA for b709925

Commits on Mar 22, 2024

• [Pallas TPU] Add support for dynamic sized (tile aligned) DMAs

  Show description for 6f0737b
  

  sharadmv

  authored and

  jax authors

  committed
  6f0737b

  Copy full SHA for 6f0737b

• Fixing StableHLO python dependencies on stablehlo:reference_api

  Show description for 6ffd55c
  

  sdasgup3

  authored and

  jax authors

  committed
  6ffd55c

  Copy full SHA for 6ffd55c

• Reverts bed4f65

  Show description for 910a31d
  

  jax authors

  committed
  910a31d

  Copy full SHA for 910a31d

• Remove Bazel workspace entries for nanobind and robin_map.

  Show description for a853539
  

  hawkinsp

  authored and

  jax authors

  committed
  a853539

  Copy full SHA for a853539

• Don't report origin_msg if any execption is raised in self._origin_msg

  Show description for 0b46341
  

  yashk2810

  authored and

  jax authors

  committed
  0b46341

  Copy full SHA for 0b46341

• Remove _maybe_device_put because jax.device_put accepts None on the device parameter

  Show description for d7e5dde
  

  yashk2810

  authored and

  jax authors

  committed
  d7e5dde

  Copy full SHA for d7e5dde

• Propagate sharding of inputs to full_like that are capable of carrying sharding as an attribute.

  Show description for 5f467b9
  

  yashk2810

  authored and

  jax authors

  committed
  5f467b9

  Copy full SHA for 5f467b9

• Remove support for CUDA 11.

  Show description for bed4f65
  

  jax authors

  committed
  bed4f65

  Copy full SHA for bed4f65

• Merge pull request jax-ml#20373 from pearu:pearu/complex-plane-tests-fix

  Show description for c82deb2
  

  jax authors

  committed
  c82deb2

  Copy full SHA for c82deb2

• Merge pull request jax-ml#20370 from jakevdp:key-reuse-flag

  Show description for 6695a85
  

  jax authors

  committed
  6695a85

  Copy full SHA for 6695a85

• Merge pull request jax-ml#20342 from ROCm:rocm-export_test-add-rocm-platform

  Show description for f6ed624
  

  jax authors

  committed
  f6ed624

  Copy full SHA for f6ed624

• Compute axis_index without creating an entire grid of device IDs.

  Show description for 44be575
  

  jax authors

  committed
  44be575

  Copy full SHA for 44be575

• Really skip exp2 in Pallas GPU tests with older jaxlib

  Show description for 2d65571
  

  superbobry

  authored and

  jax authors

  committed
  2d65571

  Copy full SHA for 2d65571

• [key reuse] rename flag to jax_debug_key_reuse

  jakevdp

  committed
  8949a63

  Copy full SHA for 8949a63

• Reverts 0e092a7

  Show description for cd79e71
  

  jax authors

  committed
  cd79e71

  Copy full SHA for cd79e71

• Update XLA dependency to use revision

  Show description for c2d9528
  

  jax authors

  committed
  c2d9528

  Copy full SHA for c2d9528

• Expose .layout on jax.Array. Also add checks in the AOT path to make sure that the input Array's layout matches the layout given to jax.jit.

  Show description for 0e092a7
  

  yashk2810

  authored and

  jax authors

  committed
  0e092a7

  Copy full SHA for 0e092a7

• Adding expectations to e2e test with expectations for mesh creation allowing splitting physical axes.

  Show description for 73aadbf
  

  jax authors

  committed
  73aadbf

  Copy full SHA for 73aadbf

• Add a fast path for Python scalars to shaped_abstractify.

  Show description for 9523547
  

  hawkinsp

  authored and

  jax authors

  committed
  9523547

  Copy full SHA for 9523547

• Merge pull request jax-ml#20236 from jakevdp:key-reuse-stack

  Show description for 07e45c3
  

  jax authors

  committed
  07e45c3

  Copy full SHA for 07e45c3

• Raise a better error message when an invalid input is passed to jit call.

  Show description for d57bb8c
  

  yashk2810

  authored and

  jax authors

  committed
  d57bb8c

  Copy full SHA for d57bb8c

• [Mosaic] Support left shifting relayouts

  Show description for 7f7e0c0
  

  tlongeri

  authored and

  jax authors

  committed
  7f7e0c0

  Copy full SHA for 7f7e0c0

Commits on Mar 21, 2024

• Enable Creating Device Mesh with Physical Axes Splits.

  Show description for 8a2ba76
  

  jax authors

  committed
  8a2ba76

  Copy full SHA for 8a2ba76

• [XLA:Python] Add a C++ implementation of flatten_one_level.

  Show description for 5532e55
  

  hawkinsp

  authored and

  jax authors

  committed
  5532e55

  Copy full SHA for 5532e55

• Expose API to control whether to fuse input computation with Pallas kernel on per input basis.

  Show description for 05e61ed
  

  jax authors

  committed
  05e61ed

  Copy full SHA for 05e61ed

• Evaluate the correctness of JAX complex functions using mpmath as a reference

  pearu

  committed
  fdb5015

  Copy full SHA for fdb5015

• Relax regex matching for DebugString output

  Show description for 4c7351f
  

  jax authors

  committed
  4c7351f

  Copy full SHA for 4c7351f

• [key reuse] print information about key reuse location

  jakevdp

  committed
  7e60331

  Copy full SHA for 7e60331

• Merge pull request jax-ml#20341 from ROCm:rocm_add_hipStreamWaitEvent

  Show description for 2848cda
  

  jax authors

  committed
  2848cda

  Copy full SHA for 2848cda

• [PJRT][IFRT] Update PJRT, IFRT, and Py executable getters to return PjRtLayouts

  Show description for 291a5cd
  

  yueshengys

  authored and

  jax authors

  committed
  291a5cd

  Copy full SHA for 291a5cd

• Attempt to eliminate flakiness for jax2tf test.

  Show description for 383ae41
  

  gnecula

  authored and

  jax authors

  committed
  383ae41

  Copy full SHA for 383ae41

Pagination

Previous

Next