Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 2, 2025

• Integrate Triton up to [2c42c562](https://github.com/openai/triton/commits/2c42c562e7c09bae5b07269482b20c6d29c40d88)

  Show description for d0cfcec
  Google-ML-Automation

  committed
  d0cfcec

  Copy full SHA for d0cfcec

Commits on May 2, 2025

• Bumped jax-triton version along with JAX/Triton requirements

  Show description for 134fcd9
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  134fcd9

  Copy full SHA for 134fcd9

Commits on May 1, 2025

• Update docs action and installation instructions

  Show description for 975a01b
  

  rdyro

  authored and

  Google-ML-Automation

  committed
  975a01b

  Copy full SHA for 975a01b

Commits on Apr 30, 2025

• fix: handle triton 3.3.0 release which released without latest changes

  Show description for f279256
  

  rdyro

  authored and

  Google-ML-Automation

  committed
  f279256

  Copy full SHA for f279256

Commits on Apr 10, 2025

• Update triton_call to use FFI APIs instead of deprecated mlir.custom_call.

  Show description for e009254
  

  dfm

  authored and

  Google-ML-Automation

  committed
  e009254

  Copy full SHA for e009254

Commits on Mar 31, 2025

• Fixed a few typos in the matmul example

  Show description for 44c9ebc
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  44c9ebc

  Copy full SHA for 44c9ebc

Commits on Mar 14, 2025

• Merge pull request #334 from superbobry:main

  Show description for 9aff066
  Google-ML-Automation

  committed
  9aff066

  Copy full SHA for 9aff066

• Support the latest Triton API changes

  Show description for 7e9af75
  superbobry

  committed
  7e9af75

  Copy full SHA for 7e9af75

Commits on Mar 13, 2025

• Add support for scratch buffers in `jax_triton`.

  Show description for 55daa46
  Google-ML-Automation

  committed
  55daa46

  Copy full SHA for 55daa46

Commits on Mar 12, 2025

• Add support for scratch buffers in `jax_triton`.

  Show description for 2d4e2eb
  

  chr1sj0nes

  authored and

  Google-ML-Automation

  committed
  2d4e2eb

  Copy full SHA for 2d4e2eb

Commits on Mar 5, 2025

• Update references to JAX's GitHub repo

  Show description for 150af1f
  

  jakeharmon8

  authored and

  Google-ML-Automation

  committed
  150af1f

  Copy full SHA for 150af1f

Commits on Jan 23, 2025

• [JAX] Use argument positions instead of names for the attributes dict.

  Show description for 5ddabf3
  

  chsigg

  authored and

  Google-ML-Automation

  committed
  5ddabf3

  Copy full SHA for 5ddabf3

• Add descriptive error messages when triton_call is used with transformations.

  Show description for 3aec7f2
  

  dfm

  authored and

  Google-ML-Automation

  committed
  3aec7f2

  Copy full SHA for 3aec7f2

Commits on Jan 21, 2025

• Integrate Triton up to [632bfc3](https://github.com/openai/triton/commits/632bfc342d3a7d63ce8b21209355139ee070d392)

  Show description for 59f5703
  

  chsigg

  authored and

  Google-ML-Automation

  committed
  59f5703

  Copy full SHA for 59f5703

Commits on Jan 15, 2025

• Integrate Triton up to [632bfc34](https://github.com/openai/triton/commits/632bfc342d3a7d63ce8b21209355139ee070d392)

  Show description for 907555f
  

  chsigg

  authored and

  Google-ML-Automation

  committed
  907555f

  Copy full SHA for 907555f

Commits on Jan 14, 2025

• Integrate Triton up to [632bfc34](https://github.com/openai/triton/commits/632bfc342d3a7d63ce8b21209355139ee070d392)

  Show description for e71d107
  

  vwbaker

  authored and

  Google-ML-Automation

  committed
  e71d107

  Copy full SHA for e71d107

Commits on Dec 23, 2024

• Integrate Triton up to [88c704e](https://github.com/openai/triton/commits/88c704e5e57542e8c34558d9a1bd2dff328f02af)

  Show description for 859cc39
  

  Moerafaat

  authored and

  Google-ML-Automation

  committed
  859cc39

  Copy full SHA for 859cc39

Commits on Dec 18, 2024

• Removed old jaxlib version guards

  Show description for c3419d6
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  c3419d6

  Copy full SHA for c3419d6

Commits on Dec 17, 2024

• Migrate from jax.core to jax.extend.core for several deprecated symbols

  Show description for 8d40b16
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  8d40b16

  Copy full SHA for 8d40b16

Commits on Dec 16, 2024

• Remove test with non-power-of-2 cluster size (unsupported by triton)

  Show description for 91a29fb
  Google-ML-Automation

  committed
  91a29fb

  Copy full SHA for 91a29fb

Commits on Dec 11, 2024

• Remove references to jax.core.raise_to_shaped

  Show description for 0a8bdd5
  

  Jake VanderPlas

  authored and

  Google-ML-Automation

  committed
  0a8bdd5

  Copy full SHA for 0a8bdd5

Commits on Oct 31, 2024

• Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)

  Show description for b00ec0c
  

  Moerafaat

  authored and

  Google-ML-Automation

  committed
  b00ec0c

  Copy full SHA for b00ec0c

Commits on Oct 23, 2024

• Merge pull request #310 from superbobry:main

  Show description for f0ed12f
  Google-ML-Automation

  committed
  f0ed12f

  Copy full SHA for f0ed12f

• Reverted the version bump

  Show description for f5c8137
  superbobry

  committed
  f5c8137

  Copy full SHA for f5c8137

• Merge pull request #308 from superbobry:maint-2

  Show description for bb0e91a
  Google-ML-Automation

  committed
  bb0e91a

  Copy full SHA for bb0e91a

Commits on Oct 22, 2024

• Bumped `jax_triton` version and updated JAX/Triton requirements

  superbobry

  committed
  76972e6

  Copy full SHA for 76972e6

• Merge pull request #298 from jax-ml:gpu-actions-testing

  Show description for aa8c729
  Google-ML-Automation

  committed
  aa8c729

  Copy full SHA for aa8c729

• Add GPU testing to Actions

  MichaelHudgins

  committed
  12a2bbb

  Copy full SHA for 12a2bbb

• Merge pull request #307 from superbobry:pallas

  Show description for 060baa2
  Google-ML-Automation

  committed
  060baa2

  Copy full SHA for 060baa2

Commits on Oct 21, 2024

• Merge pull request #305 from superbobry:maint

  Show description for 354c15a
  Google-ML-Automation

  committed
  354c15a

  Copy full SHA for 354c15a

• Assume that `jaxlib.get_serialized_metadata` is always defined

  Show description for 4d443b9
  superbobry

  committed
  4d443b9

  Copy full SHA for 4d443b9

• Nuked Pallas examples from jax_triton

  Show description for 64211a0
  superbobry

  committed
  64211a0

  Copy full SHA for 64211a0

• Merge pull request #306 from superbobry:maint-2

  Show description for cd49678
  Google-ML-Automation

  committed
  cd49678

  Copy full SHA for cd49678

• Aligned ruff settings with JAX and fixed a few warnings

  superbobry

  committed
  be5c9c2

  Copy full SHA for be5c9c2

Commits on Oct 2, 2024

• Integrate Triton up to [6fa4f504](https://github.com/openai/triton/commits/6fa4f504d61e48f7cea454ce0c1f6169907d5aaa)

  Show description for c7b8cd5
  

  vwbaker

  authored and

  Google-ML-Automation

  committed
  c7b8cd5

  Copy full SHA for c7b8cd5

Pagination

Previous

Next