Commits

Branch selector

 6695a85

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 22, 2024

• 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

• 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

• Skip PallasOpsTest.test_elementwise_exp2 on older jaxlib

  Show description for c945766
  

  superbobry

  authored and

  jax authors

  committed
  c945766

  Copy full SHA for c945766

• Merge pull request jax-ml#20359 from eltociear:patch-8

  Show description for 0a0d65a
  

  jax authors

  committed
  0a0d65a

  Copy full SHA for 0a0d65a

• Don't tree_flatten in_shardings and out_shardings each time a jit() is traced.

  Show description for 54d8bde
  

  hawkinsp

  authored and

  jax authors

  committed
  54d8bde

  Copy full SHA for 54d8bde

• Remove _python_pjit and make _cpp_pjit the only function wrapper.

  Show description for dd574cb
  

  yashk2810

  authored and

  jax authors

  committed
  dd574cb

  Copy full SHA for dd574cb

• Allow disabling checking CUDA constraints at runtime.

  Show description for 74f4846
  

  jax authors

  committed
  74f4846

  Copy full SHA for 74f4846

• Only call inspect.signature once during the initial call to jit().

  Show description for 79b1894
  

  hawkinsp

  authored and

  jax authors

  committed
  79b1894

  Copy full SHA for 79b1894

• Merge pull request jax-ml#20357 from gnecula:deprecate_host_callback

  Show description for 8bdbaf7
  

  jax authors

  committed
  8bdbaf7

  Copy full SHA for 8bdbaf7

• Refactorings to the jit implementation.

  Show description for d3e03ff
  

  hawkinsp

  authored and

  jax authors

  committed
  d3e03ff

  Copy full SHA for d3e03ff

• [pallas:triton] Implement lowering rule for broadcasted iota.

  Show description for 2bd579b
  

  chr1sj0nes

  authored and

  jax authors

  committed
  2bd579b

  Copy full SHA for 2bd579b

• [Mosaic] Enable the lightweight stable serialization format

  Show description for 590bec2
  

  apaszke

  authored and

  jax authors

  committed
  590bec2

  Copy full SHA for 590bec2

• Complemented libdevice ops with the ones from the MLIR Math dialect

  Show description for c5fa14b
  

  superbobry

  authored and

  jax authors

  committed
  c5fa14b

  Copy full SHA for c5fa14b

• Update discharge.py

  Show description for da1a2ac
  eltociear

  authored
  da1a2ac

  Copy full SHA for da1a2ac

• Add support for slicing dynamically-shaped memrefs + DMAs between them

  Show description for 7d431ad
  

  apaszke

  authored and

  jax authors

  committed
  7d431ad

  Copy full SHA for 7d431ad

• [host_callback] Deprecate the jax.experimental.host_callback module.

  gnecula

  committed
  ca59971

  Copy full SHA for ca59971

Pagination

Previous

Next