Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 3, 2024

• Merge pull request #1 from jax-ml/main

  Show description for 14f8748
  erfanzar

  authored
  14f8748

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

Commits on Oct 23, 2024

• Merge pull request jax-ml#310 from superbobry:main

  Show description for f0ed12f
  Google-ML-Automation

  committed
  f0ed12f

• Reverted the version bump

  Show description for f5c8137
  superbobry

  committed
  f5c8137

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

  Show description for bb0e91a
  Google-ML-Automation

  committed
  bb0e91a

Commits on Oct 22, 2024

• Bumped jax_triton version and updated JAX/Triton requirements

  superbobry

  committed
  76972e6

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

  Show description for aa8c729
  Google-ML-Automation

  committed
  aa8c729

• Add GPU testing to Actions

  MichaelHudgins

  committed
  12a2bbb

• Merge pull request jax-ml#307 from superbobry:pallas

  Show description for 060baa2
  Google-ML-Automation

  committed
  060baa2

Commits on Oct 21, 2024

• Merge pull request jax-ml#305 from superbobry:maint

  Show description for 354c15a
  Google-ML-Automation

  committed
  354c15a

• Assume that jaxlib.get_serialized_metadata is always defined

  Show description for 4d443b9
  superbobry

  committed
  4d443b9

• Nuked Pallas examples from jax_triton

  Show description for 64211a0
  superbobry

  committed
  64211a0

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

  Show description for cd49678
  Google-ML-Automation

  committed
  cd49678

• Aligned ruff settings with JAX and fixed a few warnings

  superbobry

  committed
  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

Commits on Sep 17, 2024

• Removed the code path for ttir being an ir.Module

  Show description for fa66b2d
  

  superbobry

  authored and

  The jax_triton Authors

  committed
  fa66b2d

Commits on Sep 16, 2024

• Integrate Triton up to [50d803cd](https://github.com/openai/triton/commits/50d803cdb4e68910ed663251100e168ea4d2519d

  Show description for 923bc6a
  

  chsigg

  authored and

  The jax_triton Authors

  committed
  923bc6a

• Merge pull request jax-ml#296 from dfm:gpu-support-error-message

  Show description for 949f07b
  

  The jax_triton Authors

  committed
  949f07b

• Merge pull request jax-ml#295 from superbobry:main

  Show description for 4ebaed0
  

  The jax_triton Authors

  committed
  4ebaed0

• Improve the error message when jaxlib is installed without GPU support.

  Show description for a5dc021
  dfm

  committed
  a5dc021

• Upgraded the sources to 3.10

  Show description for a72a159
  superbobry

  committed
  a72a159

• Merge pull request jax-ml#293 from superbobry:main

  Show description for 973e106
  

  The jax_triton Authors

  committed
  973e106

• Added pre-commit-run and enabled it on GitHub actions

  Show description for 50f4af1
  superbobry

  committed
  50f4af1

• Added an empty .pre-commit-config.yaml

  Show description for c5027d6
  

  superbobry

  authored and

  The jax_triton Authors

  committed
  c5027d6

Commits on Sep 12, 2024

• Add JAX→Triton fp8 type mapping.

  Show description for 18548c6
  

  mooskagh

  authored and

  The jax_triton Authors

  committed
  18548c6

Commits on Sep 11, 2024

• Refactors triton_kernel_call_lowering to support both cuda and rocm.

  Show description for 94fba56
  

  pschuh

  authored and

  The jax_triton Authors

  committed
  94fba56

• Copybara import of the project:

  Show description for e82c529
  

  teijeong

  authored and

  The jax_triton Authors

  committed
  e82c529

• Copybara import of the project:

  Show description for 9e7d6a9
  

  rahulbatra85

  authored and

  The jax_triton Authors

  committed
  9e7d6a9

Commits on Sep 5, 2024

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

  Show description for 67fe01a
  

  The jax_triton Authors

  committed
  67fe01a

Commits on Aug 29, 2024

• Revert due to CI breakages:

  Show description for b3a2b6c
  

  hawkinsp

  authored and

  The jax_triton Authors

  committed
  b3a2b6c

• Merge pull request jax-ml#281 from rahulbatra85:rocm_updates

  Show description for 12b8c8e
  

  The jax_triton Authors

  committed
  12b8c8e

Commits on Aug 28, 2024

• Merge pull request jax-ml#283 from hawkinsp:triton

  Show description for feb3fc3
  

  The jax_triton Authors

  committed
  feb3fc3

• Update jax-triton metadata to depend on the Triton 3.0.0 release.

  Show description for 54ed529
  hawkinsp

  committed
  54ed529

Commits on Aug 12, 2024

• ROCM updates

  

  Rahul Batra

  committed
  6742187

Commits on Jul 10, 2024

• Migrated remaining usages of pl.BlockSpec with the old argument order

  Show description for bce82cb
  

  superbobry

  authored and

  The jax_triton Authors

  committed
  bce82cb

Pagination

PreviousNext