issues
search
jax-ml
/
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
349
stars
40
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Replace Deprecated `abc` Decorators (#1)
#314
pixeeai
opened
3 weeks ago
2
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#313
copybara-service[bot]
closed
1 month ago
0
Add if conditional to GPU tests to only run on main repo
#312
MichaelHudgins
opened
1 month ago
0
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#311
copybara-service[bot]
closed
1 week ago
0
Reverted the version bump
#310
superbobry
closed
1 month ago
0
check if key_idx is already present
#309
rdyro
closed
1 month ago
1
Bumped `jax_triton` version and updated JAX/Triton requirements
#308
superbobry
closed
1 month ago
0
Nuked Pallas examples from jax_triton
#307
superbobry
closed
1 month ago
0
Aligned ruff settings with JAX and fixed a few warnings
#306
superbobry
closed
1 month ago
0
Assume that `jaxlib.get_serialized_metadata` is always defined
#305
superbobry
closed
1 month ago
0
Dropped stale jaxlib conditional
#304
superbobry
closed
1 month ago
1
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#303
copybara-service[bot]
closed
1 month ago
0
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#302
copybara-service[bot]
closed
1 week ago
0
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#300
copybara-service[bot]
opened
1 month ago
0
Integrate Triton up to [6fa4f504](https://github.com/openai/triton/commits/6fa4f504d61e48f7cea454ce0c1f6169907d5aaa)
#299
copybara-service[bot]
closed
2 months ago
0
Introduce GPU testing to JAX Triton
#298
MichaelHudgins
closed
1 month ago
0
Removed the code path for `ttir` being an `ir.Module`
#297
copybara-service[bot]
closed
2 months ago
0
Improve the error message when jaxlib is installed without GPU support.
#296
dfm
closed
2 months ago
0
Upgraded the sources to 3.10
#295
superbobry
closed
2 months ago
0
Added an empty .pre-commit-config.yaml
#294
copybara-service[bot]
closed
2 months ago
0
Added pre-commit-run and enabled it on GitHub actions
#293
superbobry
closed
2 months ago
0
Integrate Triton up to [50d803cd](https://github.com/openai/triton/commits/50d803cdb4e68910ed663251100e168ea4d2519d
#292
copybara-service[bot]
closed
2 months ago
0
Add JAX→Triton fp8 type mapping.
#291
copybara-service[bot]
closed
2 months ago
0
Refactors triton_kernel_call_lowering to support both cuda and rocm.
#290
copybara-service[bot]
closed
2 months ago
0
Now that device is used in backend_init_func, it must always be set
#289
copybara-service[bot]
closed
2 months ago
0
Copybara import of the project:
#288
copybara-service[bot]
closed
2 months ago
0
ROCM updates
#287
rahulbatra85
closed
2 months ago
4
Replace jax.lib.xla_bridge.get_backend() with jax.extend.backend.get_backend()
#286
copybara-service[bot]
closed
3 months ago
1
Revert due to CI breakages:
#285
copybara-service[bot]
closed
3 months ago
0
Stable set of dependencies?
#284
vvvm23
opened
3 months ago
1
Update jax-triton metadata to depend on the Triton 3.0.0 release.
#283
hawkinsp
closed
3 months ago
1
Integrate Triton up to [b2de88f8](https://github.com/openai/triton/commits/b2de88f89c1ff8082c92165535e48dece55da392)
#282
copybara-service[bot]
closed
2 months ago
1
ROCM updates
#281
rahulbatra85
closed
3 months ago
6
Migrated remaining usages of pl.BlockSpec with the old argument order
#280
copybara-service[bot]
closed
4 months ago
0
Migrated ``pl.BlockSpec`` uses in jax_triton to the new argument order
#279
copybara-service[bot]
closed
5 months ago
0
Integrate Triton up to [@4caedc2](https://github.com/openai/triton/commits/4caedc2a51f5c81f7b63212b9bcb76b430102913)
#278
copybara-service[bot]
closed
6 months ago
0
Removed Pallas forwarding modules from jax_triton
#277
copybara-service[bot]
closed
6 months ago
0
Clarify that jax-triton only works on GPU
#276
skye
opened
7 months ago
0
Add option to dump intermediate artifacts for jax-triton compilation
#275
copybara-service[bot]
closed
7 months ago
1
Integrate Triton up to [8e0c7b42](https://github.com/openai/triton/commits/8e0c7b425ac149c43183de966ffa423fd46e4762)
#274
copybara-service[bot]
closed
7 months ago
0
Is jax-triton replaced by Pallas?
#273
gautierronan
closed
6 months ago
2
fix compatibility issues with gpu_triton and tl_ir method access
#272
dtunai
closed
1 month ago
3
cannot import name 'ShapedArrayRef' due to deprecation in jax._src.state
#271
dtunai
opened
8 months ago
0
WIP - testing out JAX fix
#270
copybara-service[bot]
closed
8 months ago
0
Integrate Triton up to [bfb8e413](https://github.com/openai/triton/commits/bfb8e413b075583228c961bdbb65a98dc54d0868)
#269
copybara-service[bot]
closed
8 months ago
0
[jax_triton] Add parameter allowing user to compile for specific compute capability.
#268
copybara-service[bot]
closed
9 months ago
0
Upgrading to triton 3.x
#267
jbuckman
closed
8 months ago
0
Unable to install Jax-Triton
#266
bsaoptima
opened
9 months ago
1
Moved Pallas GPU lowering registartion code into a separate submodule
#265
copybara-service[bot]
closed
9 months ago
0
Import error encountered in jax_triton
#264
egg5154
opened
9 months ago
7
Next