Commits

Branch selector

 9c3365f

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 19, 2024

• Migrate shardy dialect extension to nanobind.

  Show description for 9c3365f
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  9c3365f

  Copy full SHA for 9c3365f

• Merge pull request jax-ml#25614 from jakevdp:dep-shaped-abstractify

  Show description for 3e2f2aa
  Google-ML-Automation

  committed
  3e2f2aa

  Copy full SHA for 3e2f2aa

• Update XLA dependency to use revision

  Show description for 1b385f3
  Google-ML-Automation

  committed
  1b385f3

  Copy full SHA for 1b385f3

• Merge pull request jax-ml#25606 from dfm:fft-nd

  Show description for dc8b786
  Google-ML-Automation

  committed
  dc8b786

  Copy full SHA for dc8b786

• Merge pull request jax-ml#25440 from ROCm:jax_pypi-upstream

  Show description for bad6f0f
  Google-ML-Automation

  committed
  bad6f0f

  Copy full SHA for bad6f0f

• Merge pull request jax-ml#25148 from ROCm:ci_install_sys_libs-upstream

  Show description for 25713f9
  Google-ML-Automation

  committed
  25713f9

  Copy full SHA for 25713f9

• Add experimental support for building JAX CPU and GPU wheels with GCC.

  Show description for c78ca04
  Google-ML-Automation

  committed
  c78ca04

  Copy full SHA for c78ca04

• Merge pull request jax-ml#25516 from jakevdp:jep-effver

  Show description for 49c9246
  Google-ML-Automation

  committed
  49c9246

  Copy full SHA for 49c9246

• Merge pull request jax-ml#25590 from jakevdp:fix-one-hot-float

  Show description for 8402a98
  Google-ML-Automation

  committed
  8402a98

  Copy full SHA for 8402a98

• Update documentation and add setup.py pypi bindings

  Ruturaj4

  committed
  fefebeb

  Copy full SHA for fefebeb

• Always suppress the differing_executors Hypothesis health check

  Show description for 23000a3
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  23000a3

  Copy full SHA for 23000a3

• [Mosaic GPU] Commit to using Vectors everywhere (and no Tensors).

  Show description for 3915f4a
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  3915f4a

  Copy full SHA for 3915f4a

• [jaxlib] Added a missing pytype_dep to :_triton_ext

  Show description for c4fae4a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  c4fae4a

  Copy full SHA for c4fae4a

• Add support for N-D FFTs with D>3.

  dfm

  committed
  c6131ee

  Copy full SHA for c6131ee

• [pallas:triton] Fixed a typo in a type annotation

  Show description for af7a31f
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  af7a31f

  Copy full SHA for af7a31f

• jax.nn.one_hot: deprecate non-integer inputs

  jakevdp

  committed
  8c3c441

  Copy full SHA for 8c3c441

• Remove internal uses of api_util.shaped_abstractify

  jakevdp

  committed
  5dc37d3

  Copy full SHA for 5dc37d3

• [ROCm] ci build and dockerfile changes

  Ruturaj4

  committed
  bfcace4

  Copy full SHA for bfcace4

• Merge pull request jax-ml#25595 from jakevdp:mv-shaped-abstractify

  Show description for 7680532
  Google-ML-Automation

  committed
  7680532

  Copy full SHA for 7680532

• [Mosaic TPU] Guard tests for new features by the libtpu version

  Show description for ad00ec1
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  ad00ec1

  Copy full SHA for ad00ec1

• [Pallas:TPU] Use self.pallas_call to properly handle interpret mode

  Show description for 4515949
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  4515949

  Copy full SHA for 4515949

• [Mosaic GPU] Add a new tiled layout, optimized for upcasting before WGMMA

  Show description for 006c65d
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  006c65d

  Copy full SHA for 006c65d

• [Mosaic GPU] Replace the dialect's layout enum with layouts holding the proper

  Show description for 66ad208
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  66ad208

  Copy full SHA for 66ad208

• [Mosaic:TPU][NFC] Small cleanup of extui rule in apply-vector-layout

  Show description for 8188c57
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  8188c57

  Copy full SHA for 8188c57

• [Mosaic GPU][NFC] Move up the definition of ThreadSemantics.

  Show description for 16f0f97
  

  dimitar-asenov

  authored and

  Google-ML-Automation

  committed
  16f0f97

  Copy full SHA for 16f0f97

• Add lax.composite primitive

  Show description for d206cc3
  

  ghpvnist

  authored and

  Google-ML-Automation

  committed
  d206cc3

  Copy full SHA for d206cc3

• Refactor: move shaped_abstractify to core

  jakevdp

  committed
  676070f

  Copy full SHA for 676070f

• [Pallas] Fix lowering tests for reduction ops

  Show description for 613a0cd
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  613a0cd

  Copy full SHA for 613a0cd

• Merge pull request jax-ml#25593 from mattjj:ref-errors-4

  Show description for f65eced
  Google-ML-Automation

  committed
  f65eced

  Copy full SHA for f65eced

• Account for tokens in allow_spmd_sharding_propagation_to_parameters and allow_spmd_sharding_propagation_to_output compile options

  Show description for 9041b02
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  9041b02

  Copy full SHA for 9041b02

• Merge pull request jax-ml#25596 from mattjj:whitespace

  Show description for 46b18d2
  Google-ML-Automation

  committed
  46b18d2

  Copy full SHA for 46b18d2

• add mutable array ref error checks to scan

  mattjj

  committed
  e528562

  Copy full SHA for e528562

• [Mosaic TPU] Support i32 vector multi reduction except cross lane.

  Show description for 3a5c4da
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  3a5c4da

  Copy full SHA for 3a5c4da

• [Mosaic] Improve error verbosity of tpu.memref_slice verification

  Show description for 6bcec91
  

  naummo

  authored and

  Google-ML-Automation

  committed
  6bcec91

  Copy full SHA for 6bcec91

Commits on Dec 18, 2024

• remove whitespace from distributed arrays doc

  mattjj

  committed
  c657682

  Copy full SHA for c657682

Pagination

Previous

Next