Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 27, 2024

• Update XLA dependency to use revision

  Show description for 7ab61b7
  Google-ML-Automation

  committed
  7ab61b7

Commits on Dec 26, 2024

• Update XLA dependency to use revision

  Show description for 6dbda90
  Google-ML-Automation

  committed
  6dbda90

Commits on Dec 25, 2024

• Update XLA dependency to use revision

  Show description for 42a0d55
  Google-ML-Automation

  committed
  42a0d55

• [AutoPGLE] Explicitly disable command buffers when profiler is used.

  Show description for b6aead6
  Google-ML-Automation

  committed
  b6aead6

Commits on Dec 24, 2024

• Update XLA dependency to use revision

  Show description for 64511a1
  Google-ML-Automation

  committed
  64511a1

• [pallas:mosaic_gpu] Addressed a todo in broadcasted_iota lowering

  Show description for 44333e1
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  44333e1

• Merge pull request #25672 from jakevdp:finalize-dep

  Show description for 4eff131
  Google-ML-Automation

  committed
  4eff131

• Merge pull request #25674 from jax-ml:dependabot/github_actions/actions/upload-artifact-4.5.0

  Show description for fa9c7ed
  Google-ML-Automation

  committed
  fa9c7ed

Commits on Dec 23, 2024

• [Mosaic:TPU] In infer ext rule, avoid assigning offsets outside of dst first tile

  Show description for 4452960
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  4452960

• Switch mlir bindings from pybind11 to nanobind

  Show description for b8091a4
  Google-ML-Automation

  committed
  b8091a4

• [Mosaic:TPU] Vreg-slice-aligned offset changes with scratch retiling

  Show description for 3c79b98
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  3c79b98

• Merge pull request #25669 from jakevdp:undep

  Show description for c57b49c
  Google-ML-Automation

  committed
  c57b49c

• Finalize deprecation of some symbols from jax.lib.xla_client

  jakevdp

  committed
  40fe4b8

• Bump actions/upload-artifact from 4.4.3 to 4.5.0

  Show description for 7da753e
  dependabot[bot]

  authored
  7da753e

• Merge pull request #25673 from jakevdp:changelog

  Show description for 1087623
  Google-ML-Automation

  committed
  1087623

• changelog: link to api compatibility & python version docs

  jakevdp

  committed
  c206ae7

• Update XLA dependency to use revision

  Show description for 23965b7
  Google-ML-Automation

  committed
  23965b7

• Merge pull request #25651 from jakevdp:combine-abstractify

  Show description for 51b5102
  Google-ML-Automation

  committed
  51b5102

• Internal: use a single registry for abstractify APIs

  jakevdp

  committed
  ccc3a29

• Merge pull request #25662 from liblaf:main

  Show description for 6c85e54
  Google-ML-Automation

  committed
  6c85e54

• Merge pull request #24607 from kaixih:support_head_size_256

  Show description for 704185e
  Google-ML-Automation

  committed
  704185e

• Use the right include for gmock and gtest

  Show description for 68ec202
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  68ec202

• Remove casting from jax.nn.one_hot

  Show description for cb10710
  jakevdp

  committed
  cb10710

• Merge pull request #25667 from superbobry:main

  Show description for 76a51f8
  Google-ML-Automation

  committed
  76a51f8

• [mosaic_gpu] Include Mosaic GPU dialect fiels into jaxlib

  superbobry

  committed
  8987867

• [pallas:triton] Add support for lowering int4 load.

  Show description for 83e60a9
  

  chr1sj0nes

  authored and

  Google-ML-Automation

  committed
  83e60a9

• [pallas:mosaic_gpu] Reduced duplication between _ensure_fa and _ensure_ir_value

  Show description for a51d627
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  a51d627

• [pallas:mosaic_gpu] Updated the lowering following the changes in in Mosaic GPU internals

  Show description for 3e7f481
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  3e7f481

• Fix a typo in documentation for pinv function.

  liblaf

  committed
  75b5654

Commits on Dec 22, 2024

• Move ragged tests under a new class.

  Show description for 38747a7
  

  ghpvnist

  authored and

  Google-ML-Automation

  committed
  38747a7

• [Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering.

  Show description for 1719986
  Google-ML-Automation

  committed
  1719986

Commits on Dec 21, 2024

• Update XLA dependency to use revision

  Show description for d28b0da
  Google-ML-Automation

  committed
  d28b0da

• Merge pull request #25650 from jakevdp:view-int4

  Show description for 1c0dee8
  Google-ML-Automation

  committed
  1c0dee8

• Move output path to be inside the wheel build command execution loop

  Show description for 0159bea
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  0159bea

• [JAX] Add a test using inputs with different device orders for a single colocated Python call

  Show description for 0e0fc0a
  

  hyeontaek

  authored and

  Google-ML-Automation

  committed
  0e0fc0a

Pagination

PreviousNext