Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 21, 2024

• Nuked Pallas examples from jax_triton

  Show description for 64211a0
  superbobry

  committed
  64211a0

Commits on Sep 16, 2024

• Upgraded the sources to 3.10

  Show description for a72a159
  superbobry

  committed
  a72a159

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

  Show description for 50f4af1
  superbobry

  committed
  50f4af1

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

Commits on Jul 2, 2024

• Migrated pl.BlockSpec uses in jax_triton to the new argument order

  Show description for 7965cfb
  

  superbobry

  authored and

  The jax_triton Authors

  committed
  7965cfb

Commits on Jan 30, 2024

• Integrate Triton up to [9f816a7b](https://github.com/openai/triton/commits/9f816a7b984ce20ec684866f9c8bb5ffd49e2500)

  Show description for 28ad476
  

  gflegar

  authored and

  The jax_triton Authors

  committed
  28ad476

Commits on Aug 9, 2023

• [JAX] Replace uses of jnp.array in types with jnp.ndarray.

  Show description for a25b1ba
  

  hawkinsp

  authored and

  The jax_triton Authors

  committed
  a25b1ba

Commits on Jul 19, 2023

• Update the code to not use the deprecated Chex assertions.

  Show description for bbf4287
  

  hbq1

  authored and

  The jax_triton Authors

  committed
  bbf4287

Commits on Apr 14, 2023

• Merge pull request #50 from jax-ml:mlir

  Show description for ab27a08
  

  The jax_triton Authors

  committed
  ab27a08

Commits on Mar 31, 2023

• Make it explicit that jax_triton depends on absl-py.

  Show description for 3285e5b
  

  chr1sj0nes

  authored and

  The jax_triton Authors

  committed
  3285e5b

Commits on Mar 30, 2023

• Cap auto-tuning at 100 iterations.

  Show description for 0c16243
  

  chr1sj0nes

  authored and

  The jax_triton Authors

  committed
  0c16243

Commits on Mar 21, 2023

• Rebase onto Triton-MLIR

  sharadmv

  committed
  de1073d

Commits on Feb 17, 2023

• Add experimental fused layer norm backward pass.

  Show description for 081d4b1
  

  zhangqiaorjc

  and

  sharadmv

  committed
  081d4b1

Commits on Feb 9, 2023

• Add fused attention op

  Show description for 7384900
  

  sharadmv

  and

  zhangqiaorjc

  committed
  7384900

Commits on Dec 1, 2022

• Upgrade pallas_call to include block specs

  sharadmv

  committed
  e138d3e

Commits on Nov 21, 2022

• Merge pull request #43 from jax-ml:lstm

  Show description for b7c71c7
  

  The jax_triton Authors

  committed
  b7c71c7

• Add Pallas LSTM example

  sharadmv

  committed
  26e5629

Commits on Nov 19, 2022

• Add basic pretty printing for pl.load/store

  sharadmv

  committed
  88fda59

Commits on Nov 17, 2022

• Add blocksparse example and while lowering

  Show description for 7337b17
  

  sharadmv

  and

  jakevdp

  committed
  7337b17

Commits on Nov 16, 2022

• Add tests for jvp of pallas_call

  sharadmv

  committed
  0ecc1da

Commits on Nov 15, 2022

• Add layer norm pallas example

  sharadmv

  committed
  9eb5844

• Add experimental pallas interpreter for verifying kernel correctness

  sharadmv

  committed
  65325fe

Commits on Nov 8, 2022

• Support None in pl.dslice

  sharadmv

  committed
  95dc491

• Copybara import of the project:

  Show description for 01b7f01
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  01b7f01

Commits on Nov 2, 2022

• Add templating example

  sharadmv

  committed
  ee39fb3

• Merge pull request #18 from jax-ml:grid-map

  Show description for cee0269
  

  The jax_triton Authors

  committed
  cee0269

• Basic grid map + flashattn

  Show description for 9ecc83a
  

  sharadmv

  and

  hawkinsp

  committed
  9ecc83a

Commits on Nov 1, 2022

• Basic broadcasting support for pl.load/store

  sharadmv

  committed
  6c18a11

Commits on Oct 29, 2022

• Copybara import of the project:

  Show description for c6314c0
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  c6314c0

Commits on Oct 26, 2022

• Add examples back

  Show description for 56e4db1
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  56e4db1

• Internal change

  Show description for de197fd
  sharadmv

  committed
  de197fd

Commits on Sep 13, 2022

• Refactor triton call

  sharadmv

  committed
  47498b3

Commits on Aug 1, 2022

• Initial code commit

  Show description for 332e66f
  

  3 peoplecommitted
  332e66f

Loading