Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 16, 2024

• Merge pull request #19793 from mattjj:remat-in-hlo-metadata

  Show description for 86db2de
  

  jax authors

  committed
  86db2de

  Copy full SHA for 86db2de

• Merge pull request #19819 from mattjj:scan-dont-traverse-body-jaxpr-in-lowering

  Show description for 330afdc
  

  jax authors

  committed
  330afdc

  Copy full SHA for 330afdc

• prototype 'remat' in hlo metadata

  mattjj

  committed
  08dfb11

  Copy full SHA for 08dfb11

• [scan] don't traverse body jaxpr in lowering

  Show description for 5ead7a6
  mattjj

  committed
  5ead7a6

  Copy full SHA for 5ead7a6

• Update XLA dependency to use revision

  Show description for ff3247e
  

  jax authors

  committed
  ff3247e

  Copy full SHA for ff3247e

• Add a benchmark measuring device_put's speed for a 4GB input array

  Show description for 0b542ff
  

  yashk2810

  authored and

  jax authors

  committed
  0b542ff

  Copy full SHA for 0b542ff

• [JAX] Add an option subset_by_index that allows computing a contiguous subset of singular components from svd.

  Show description for 7e7094c
  

  jax authors

  committed
  7e7094c

  Copy full SHA for 7e7094c

• Merge pull request #19837 from jakevdp:key-reuse-clerrs

  Show description for 0203d15
  

  jax authors

  committed
  0203d15

  Copy full SHA for 0203d15

Commits on Feb 15, 2024

• [key reuse] simplify key reuse logic through context-free jaxpr evaluation

  Show description for 8eab599
  jakevdp

  committed
  8eab599

  Copy full SHA for 8eab599

• [Mosaic] In apply_vector_layout.cc, check layout validity when reading the attribute

  Show description for 243e7ed
  

  tlongeri

  authored and

  jax authors

  committed
  243e7ed

  Copy full SHA for 243e7ed

• Update the cuda 12 dependencies since we upgraded to cuda 12.3

  Show description for f125509
  

  yashk2810

  authored and

  jax authors

  committed
  f125509

  Copy full SHA for f125509

• Micro-optimization: use .dtype rather than dtypes.result_type when forming MLIR ndarray constants.

  Show description for 4834423
  

  hawkinsp

  authored and

  jax authors

  committed
  4834423

  Copy full SHA for 4834423

• Don't call inspect.signature() each time we trace a jit().

  Show description for b5e4ba4
  

  hawkinsp

  authored and

  jax authors

  committed
  b5e4ba4

  Copy full SHA for b5e4ba4

• Partial rollback xla translation APIs that were removed in 0.4.24 release

  Show description for 8888006
  

  yashk2810

  authored and

  jax authors

  committed
  8888006

  Copy full SHA for 8888006

• [XLA:Mosaic] Prevent generating rotate op if both shift and stride are zeros.

  Show description for a37f2d4
  

  bythew3i

  authored and

  jax authors

  committed
  a37f2d4

  Copy full SHA for a37f2d4

• [Mosaic] In apply_vector_layout, prefer returning failure over CHECKs and add macros for this

  Show description for 72c5aea
  

  tlongeri

  authored and

  jax authors

  committed
  72c5aea

  Copy full SHA for 72c5aea

• Merge pull request #19831 from jakevdp:key-reuse-simplify

  Show description for 210d46f
  

  jax authors

  committed
  210d46f

  Copy full SHA for 210d46f

• Don't recompute abstract eval rules when inlining a jit jaxpr.

  Show description for 885e8a2
  

  hawkinsp

  authored and

  jax authors

  committed
  885e8a2

  Copy full SHA for 885e8a2

• [jax:pallas:triton] Fix Triton lowering rule for `broadcast_to` with constant input.

  Show description for ccb8863
  

  chr1sj0nes

  authored and

  jax authors

  committed
  ccb8863

  Copy full SHA for ccb8863

• [XLA:Mosaic] Fix dim check in concatenate helper function.

  Show description for 273b5e2
  

  bythew3i

  authored and

  jax authors

  committed
  273b5e2

  Copy full SHA for 273b5e2

• Add Pallas attention kernel for GPU serving.

  Show description for 9fcf9e5
  

  2 peopleauthored and

  jax authors

  committed
  9fcf9e5

  Copy full SHA for 9fcf9e5

• [key reuse] remove alternate simple implementation

  jakevdp

  committed
  8284c16

  Copy full SHA for 8284c16

• To avoid the inconsistency between process_index and process_id, replace backend.process_index with distributed.global_state.process_id in Jax compilation _cache_write function.

  Show description for 3708336
  

  jax authors

  committed
  3708336

  Copy full SHA for 3708336

• Merge pull request #19804 from gnecula:poly_speed2

  Show description for a23ae0d
  

  jax authors

  committed
  a23ae0d

  Copy full SHA for a23ae0d

• [shape_polyO] Performance improvements for symbolic dimension manipulations (step 2)

  Show description for eb9caf0
  gnecula

  committed
  eb9caf0

  Copy full SHA for eb9caf0

• Remove code that guards against int4 not being available.

  Show description for c55f187
  

  hawkinsp

  authored and

  jax authors

  committed
  c55f187

  Copy full SHA for c55f187

• Merge pull request #19776 from gnecula:poly_split

  Show description for 6a98382
  

  jax authors

  committed
  6a98382

  Copy full SHA for 6a98382

• [shape_poly] Add support for jnp.split

  gnecula

  committed
  18698a1

  Copy full SHA for 18698a1

• Merge pull request #19780 from gnecula:fix_jvp_astype

  Show description for ddb178a
  

  jax authors

  committed
  ddb178a

  Copy full SHA for ddb178a

• Fix JVP rule for convert_element_type to integer types.

  Show description for ddc2487
  gnecula

  committed
  ddc2487

  Copy full SHA for ddc2487

• Update XLA dependency to use revision

  Show description for bfd29f6
  

  jax authors

  committed
  bfd29f6

  Copy full SHA for bfd29f6

• Check if the Triton dialect bindings are available in lib/triton.py

  Show description for b4c8b0e
  

  superbobry

  authored and

  jax authors

  committed
  b4c8b0e

  Copy full SHA for b4c8b0e

• Merge pull request #19813 from jakevdp:reuse-key

  Show description for 7510437
  

  jax authors

  committed
  7510437

  Copy full SHA for 7510437

Commits on Feb 14, 2024

• Remove the access of `mesh.devices` from shard_map in tracing mode. It's okay to access it in eager mode

  Show description for da32609
  

  yashk2810

  authored and

  jax authors

  committed
  da32609

  Copy full SHA for da32609

• Update XLA dependency to use revision

  Show description for d844867
  

  jax authors

  committed
  d844867

  Copy full SHA for d844867

Pagination

Previous

Next