Code

Add a Softmax Triton implementation.#154

Code

Open

copybara-service[bot] wants to merge 1 commit intomainjax-ml/jax-triton:mainfrom test_535380875jax-ml/jax-triton:test_535380875Copy head branch name to clipboard

+155

Commits

Commits on May 25, 2023

• Add a Softmax Triton implementation.

  Show description for 27a1a5e
  

  bchetioui

  authored and

  The jax_triton Authors

  committed
  27a1a5e

  Copy full SHA for 27a1a5e