Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
mirrors
ROCmSoftwarePlatform
jax
Repository
Branches
Overview
Active
Stale
All
pallas_exp2
7814c32a
·
Add exp2 lowering rule
·
Oct 10, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.16-w-generic-triton
e8dc2e70
·
[ROCm]: Add jax-triton support for ROCm
·
Oct 10, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.16-w-atten-update
f5e05b52
·
Increase block size to 128 x 128
·
Oct 10, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
pallas_attn_block_size
36939dd6
·
Set num_stages to 1
·
Oct 09, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.16
e05a86a3
·
[ROCm]: Update build script
·
Oct 09, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
dependabot/github_actions/styfle/cancel-workflow-action-0.12.0
3c0e9164
·
Bump styfle/cancel-workflow-action from 0.11.0 to 0.12.0
·
Oct 09, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jax-triton
5402a769
·
[ROCm]: Add jax-triton support for ROCm
·
Sep 23, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.14-build-wheel
aa82077f
·
[ROCm]: Pin build package version
·
Sep 21, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
dependabot/github_actions/actions/checkout-4
a65bf48d
·
Bump actions/checkout from 3 to 4
·
Sep 11, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm_build_updates_1
4091ac64
·
[ROCm]: Fix duplicate deps include
·
Sep 09, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.15
cf073db4
·
[ROCm]: More Fixes
·
Sep 08, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jax-triton-4ac2bdc2b1d71ec0010412a3248ad567f145f773
8279a13f
·
[ROCm]: Update fused attention test tolerances
·
Sep 08, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
jax-triton
6f7afead
·
[ROCm]: Port Jax-Triton API to ROCm
·
Sep 07, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.14-jax-triton
ef696dc3
·
[ROCm]: Port Jax-Triton API to ROCm
·
Aug 23, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.14-hipblaslt-dbg
7e7cb325
·
Hipblaslt build flag dbg
·
Aug 22, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.14
01fa9ada
·
[ROCm]: Disable sparse tests
·
Aug 18, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.14-rocm56
d7c149e5
·
[ROCm]: CI scripts update to handle ROCm versions
·
Aug 15, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.13-jax-triton-rocm56
5e51b39c
·
[ROCm]: Update docker file to ROCm5.6
·
Aug 02, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.13-rocm56
9d98e218
·
[ROCm]: Update docker file to ROCm5.6
·
Aug 01, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rocm-jaxlib-v0.4.9-rocm56
fc46bbaa
·
ROCM: Update dockerfile to ROCm5.6
·
Aug 01, 2023
Compare
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
2
3
4
5
…
15
Next