Pallas docs#146

sharadmv wants to merge 3 commits intomainjax-ml/jax-triton:mainfrom

pallas-docsjax-ml/jax-triton:pallas-docsCopy head branch name to clipboard