Commits

Branch selector

 0d2dfea

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 27, 2024

• Add a private set_mesh API to enter into sharding_in_types mode. This is how users will enable sharding in types mode (with correct axis types set too but that doesn't work yet).

  Show description for 0d2dfea
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  0d2dfea

  Copy full SHA for 0d2dfea

• Add new CI script to run Bazel GPU (non-RBE) jobs

  Show description for 1372669
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  1372669

  Copy full SHA for 1372669

• Add a check for return codes of executor.run so that we propagate error codes correctly

  Show description for c6866d0
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  c6866d0

  Copy full SHA for c6866d0

• Install git before actions/checkout

  Show description for afcef67
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  afcef67

  Copy full SHA for afcef67

Commits on Nov 26, 2024

• Move tsl/platform/{build_config,build_config_root,rules_cc}.bzl to xla/tsl/platform

  Show description for 10fdee3
  

  ddunl

  authored and

  Google-ML-Automation

  committed
  10fdee3

  Copy full SHA for 10fdee3

• Update XLA dependency to use revision

  Show description for 9c42379
  Google-ML-Automation

  committed
  9c42379

  Copy full SHA for 9c42379

• [JAX] Add Python binding for building a colocated Python program

  Show description for bbaec6e
  

  hyeontaek

  authored and

  Google-ML-Automation

  committed
  bbaec6e

  Copy full SHA for bbaec6e

• Fix a weird interaction with set_local and empty tuples passed to it.

  Show description for 6763fcf
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  6763fcf

  Copy full SHA for 6763fcf

• Update XLA dependency to use revision

  Show description for e453fa1
  

  belitskiy

  authored and

  Google-ML-Automation

  committed
  e453fa1

  Copy full SHA for e453fa1

• [AutoPGLE] Fix pgle test after removing pjit cache.

  Show description for 92e18e6
  Google-ML-Automation

  committed
  92e18e6

  Copy full SHA for 92e18e6

• [Pallas TPU] Better error message for lowering sp.broadcast_to_p

  Show description for dc11d40
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  dc11d40

  Copy full SHA for dc11d40

• [AutoPGLE] Explicitly ignore host callback pointers

  Show description for 231967f
  Google-ML-Automation

  committed
  231967f

  Copy full SHA for 231967f

• [mosaic_gpu] Fixed unbounded recursion in FragmentedArray._pointwise

  Show description for b6566c8
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b6566c8

  Copy full SHA for b6566c8

• Use xla_extension_version instead of jaxlib_version

  Show description for 16a5607
  Google-ML-Automation

  committed
  16a5607

  Copy full SHA for 16a5607

• Merge pull request jax-ml#25084 from ROCm:ci_rocm_version

  Show description for 024e331
  Google-ML-Automation

  committed
  024e331

  Copy full SHA for 024e331

• [mgpu] Pointwise min

  Show description for f828f2d
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  f828f2d

  Copy full SHA for f828f2d

• Create a null_mesh_context internal context manager to handle null contexts properly.

  Show description for 627debc
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  627debc

  Copy full SHA for 627debc

• Add sharding argument to reshape since it also takes a shape argument for the output shape

  Show description for 59e13f8
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  59e13f8

  Copy full SHA for 59e13f8

• [mgpu/pallas_mgpu] Pointwise tanh support

  Show description for c5dc980
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  c5dc980

  Copy full SHA for c5dc980

• [pallas_mgpu] Allow trees (eg tuples) to be returned from cond_p expressions.

  Show description for ef7df1a
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  ef7df1a

  Copy full SHA for ef7df1a

Commits on Nov 25, 2024

• Update XLA dependency to use revision

  Show description for ebea435
  Google-ML-Automation

  committed
  ebea435

  Copy full SHA for ebea435

• Add new CI scripts for building JAX artifacts

  Show description for f7e9f62
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  f7e9f62

  Copy full SHA for f7e9f62

• Merge pull request jax-ml#25041 from dfm:ffi-example-refactor

  Show description for 788f493
  Google-ML-Automation

  committed
  788f493

  Copy full SHA for 788f493

• Re-factor build CLI to a subcommand based approach

  Show description for 6761512
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  6761512

  Copy full SHA for 6761512

• [SDY] remove TODO for enabling Layouts for Shardy post cl/697715276.

  Show description for f22bafa
  

  Varcho

  authored and

  Google-ML-Automation

  committed
  f22bafa

  Copy full SHA for f22bafa

• drop compute capability check

  Show description for 95029ab
  Google-ML-Automation

  committed
  95029ab

  Copy full SHA for 95029ab

• [Mosaic GPU] Support batch dimensions in FA3 MGPU kernel.

  Show description for 107bc96
  

  justinjfu

  authored and

  Google-ML-Automation

  committed
  107bc96

  Copy full SHA for 107bc96

• Remove _pjit_lower_cached cache. We can simplify the caching of jit as we have downstream caches and a cpp cache too.

  Show description for deab6fb
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  deab6fb

  Copy full SHA for deab6fb

• [SDY] Enable test_pjit_array_multi_input_multi_output since Shardy conflict resolution is now complete.

  Show description for 066859e
  

  Varcho

  authored and

  Google-ML-Automation

  committed
  066859e

  Copy full SHA for 066859e

• [SDY] enable cpu_shardy for JAX shard_alike test.

  Show description for bb1024f
  

  Varcho

  authored and

  Google-ML-Automation

  committed
  bb1024f

  Copy full SHA for bb1024f

• Add abstract mesh context manager to trace_context in the fallback path too (which will be deleted after jax 0.4.36 release)

  Show description for c35f8b2
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  c35f8b2

  Copy full SHA for c35f8b2

• Automated Code Change

  Show description for aa05dc0
  Google-ML-Automation

  committed
  aa05dc0

  Copy full SHA for aa05dc0

• [ROCm] Add rocm version information

  Ruturaj4

  committed
  e8934b9

  Copy full SHA for e8934b9

• [Mosaic GPU] Simplify logic for pointwise splat operands

  Show description for 914600a
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  914600a

  Copy full SHA for 914600a

• Refactor FFI examples to consolidate several examples into one submodule.

  dfm

  committed
  84a9cba

  Copy full SHA for 84a9cba

Pagination

Previous

Next