Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 11, 2024

• Merge pull request jax-ml#23572 from rajasekharporeddy:testbranch2

  Show description for 0faac82
  Google-ML-Automation

  committed
  0faac82

  Copy full SHA for 0faac82

• Flip jax_pmap_no_rank_reduction by default to True.

  Show description for bf2237a
  

  pschuh

  authored and

  Google-ML-Automation

  committed
  bf2237a

  Copy full SHA for bf2237a

• Enable the enhanced TPU launch barrier on all TPU generations.

  Show description for e64de98
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  e64de98

  Copy full SHA for e64de98

• Disable complex arctan test in mnegj.real part of plane.

  Show description for 9a8dc6b
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  9a8dc6b

  Copy full SHA for 9a8dc6b

• Merge pull request jax-ml#23578 from hawkinsp:postrelease

  Show description for e86d77c
  Google-ML-Automation

  committed
  e86d77c

  Copy full SHA for e86d77c

• Update XLA dependency to use revision

  Show description for 7ce8ff2
  Google-ML-Automation

  committed
  7ce8ff2

  Copy full SHA for 7ce8ff2

• Add link to XLA documentation for building JAX with CUDA from sources.

  Show description for 7c8508e
  Google-ML-Automation

  committed
  7c8508e

  Copy full SHA for 7c8508e

• Update version numbers after v0.4.32 release.

  hawkinsp

  committed
  3e81ae5

  Copy full SHA for 3e81ae5

• Prepare for v0.4.32 release.

  hawkinsp

  committed
  1594d2f

  Copy full SHA for 1594d2f

• Merge pull request jax-ml#22702 from keshavb96:rm_custom_partitioning_pointer

  Show description for 859188b
  Google-ML-Automation

  committed
  859188b

  Copy full SHA for 859188b

• Merge pull request jax-ml#23566 from puct9:main

  Show description for c708b7c
  Google-ML-Automation

  committed
  c708b7c

  Copy full SHA for c708b7c

• [Pallas] Fix TPU large array indexing tests.

  Show description for 10057eb
  

  justinjfu

  authored and

  Google-ML-Automation

  committed
  10057eb

  Copy full SHA for 10057eb

• Improve docs for jax.numpy arithmetic comparison operations

  rajasekharporeddy

  committed
  3a0ee84

  Copy full SHA for 3a0ee84

• Merge pull request jax-ml#23546 from jakevdp:stack-doc

  Show description for 2efa713
  Google-ML-Automation

  committed
  2efa713

  Copy full SHA for 2efa713

• Improve docs for jnp.stack, jnp.concat, & related functions

  jakevdp

  committed
  c33a954

  Copy full SHA for c33a954

• Merge pull request jax-ml#23545 from mattjj:readme-title-tweak

  Show description for 4839b00
  Google-ML-Automation

  committed
  4839b00

  Copy full SHA for 4839b00

• Make sure to call the superclass' __init__() on a newly created instance in PositionalSharding._remake().

  Show description for ed849ff
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  ed849ff

  Copy full SHA for ed849ff

• Relax test tolerance in pinv test to fix a CI failure on Windows CPU.

  Show description for 2bd1fde
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  2bd1fde

  Copy full SHA for 2bd1fde

• Merge pull request jax-ml#23415 from kaixih:key_value_seq_lengths

  Show description for e869a9d
  Google-ML-Automation

  committed
  e869a9d

  Copy full SHA for e869a9d

• Internal change

  Show description for ea68f45
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  ea68f45

  Copy full SHA for ea68f45

• Fix typo in jax.typing module doc

  puct9

  authored
  e5107c1

  Copy full SHA for e5107c1

• Disable a pallas export compatibility test that fails on TPU v6e.

  Show description for 49dd6ed
  

  hawkinsp

  authored and

  jax authors

  committed
  49dd6ed

  Copy full SHA for 49dd6ed

• Update users of jax.tree.map() to be more careful about how they handle Nones.

  Show description for 808003b
  

  hawkinsp

  authored and

  jax authors

  committed
  808003b

  Copy full SHA for 808003b

• Squashed commit of the following:

  Show description for 7c660c4
  keshavb96

  committed
  7c660c4

  Copy full SHA for 7c660c4

• [Pallas] Implement tiled and swizzled Memref loads for Mosaic GPU via "GPUBlockSpec"

  Show description for e3c4b20
  

  justinjfu

  authored and

  jax authors

  committed
  e3c4b20

  Copy full SHA for e3c4b20

Commits on Sep 10, 2024

• [Pallas] Disable win32 gpu_ops_test.

  Show description for c659dc9
  

  justinjfu

  authored and

  jax authors

  committed
  c659dc9

  Copy full SHA for c659dc9

• Merge pull request jax-ml#23549 from pschuh:docs-update

  Show description for 14b8625
  

  jax authors

  committed
  14b8625

  Copy full SHA for 14b8625

• Reverts 5e4250e

  Show description for 0b04dd0
  

  yashk2810

  authored and

  jax authors

  committed
  0b04dd0

  Copy full SHA for 0b04dd0

• Removed a sneaky comma in Pallas Mosaic GPU lowering

  Show description for e7b261c
  

  superbobry

  authored and

  jax authors

  committed
  e7b261c

  Copy full SHA for e7b261c

• Prepare for jax 0.4.32 release

  Show description for 5e4250e
  

  yashk2810

  authored and

  jax authors

  committed
  5e4250e

  Copy full SHA for 5e4250e

• Clean up JAX backend for all backends to avoid dangling PyClient references.

  Show description for 4957ab9
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  4957ab9

  Copy full SHA for 4957ab9

• Update the docs for conv_general_dilated to clarify 'W' 'H'.

  pschuh

  committed
  030b6c6

  Copy full SHA for 030b6c6

• [Pallas] Simplify lowering and fix the test for lax.erf_inv_p

  Show description for 46bcb1e
  

  ayaka14732

  authored and

  jax authors

  committed
  46bcb1e

  Copy full SHA for 46bcb1e

• Update XLA dependency to use revision

  Show description for 8681bf6
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  8681bf6

  Copy full SHA for 8681bf6

• Merge pull request jax-ml#23443 from selamw1:kron_outer_docstring

  Show description for deda649
  

  jax authors

  committed
  deda649

  Copy full SHA for deda649

Pagination

Previous

Next