ROCm/jax

Pull Requests Commits

Reduce pytest threads

charleshofer committed 324 days ago

c20bb5b4

refactor: move lax_numpy window functions into their own file

jakevdp committed 324 days ago

17215177

Merge pull request #26377 from mattjj:maintain-mutable-array-sharding

Google-ML-Automation committed 324 days ago

c0ba3626

Merge pull request #26376 from jakevdp:array-creation

Google-ML-Automation committed 324 days ago

ec477634

[pallas:mosaic_gpu] Fixed a bug in how `delay_release` is handled in `emit_pipeline`

superbobry committed 324 days ago

e5058079

[mutable-arrays] persist shardings through xla computations

mattjj committed 324 days ago

719031c1

Merge pull request #26409 from hawkinsp:fstring

Google-ML-Automation committed 324 days ago

7ffb613b

Merge pull request #26383 from jakevdp:jnp-sorting

Google-ML-Automation committed 324 days ago

3b470b95

DOC: make clear that printoptions are NumPy aliases

jakevdp committed 324 days ago

08563842

Fix a missing "f" on an f-string.

hawkinsp committed 324 days ago

d01520c6

Merge pull request #25810 from ROCm:gh-9948-add-gpu-ci-upstream

Google-ML-Automation committed 324 days ago

1ca8807d

jax.lax: better docs for hyperbolic trig functions

jakevdp committed 324 days ago

311e6683

[xla:python] Add a mechanism for "batch partitioning" of FFI calls.

dfm committed 324 days ago

c521bc62

Remove the unused cu_cholesky_update kernel in favor of the FFI version.

dfm committed 324 days ago

5bc17f7e

refactor: move sorting ops out of lax_numpy

jakevdp committed 324 days ago

d3b3cd36

Merge pull request #26344 from Cjkkkk:disable_head_256_on_bw

Google-ML-Automation committed 324 days ago

e56c7dc5

[pallas:triton] Really revert to the lowering using Triton IR

superbobry committed 324 days ago

35351f95

Merge pull request #26368 from ROCm:fix-rocm-pallas-lowerings

Google-ML-Automation committed 324 days ago

6ad38af4

[Mosaic GPU] Add tests for various tcgen05.mma configurations

apaszke committed 324 days ago

3b5e91b8

Add check to `lax.composite` to prevent `DynamicJaxprTracer` type errors.

ghpvnist committed 324 days ago

2828bce2

Update XLA dependency to use revision

Google-ML-Automation committed 324 days ago

4c402b7b

[better_errors] Continue adding debug info to Jaxprs (step 5)

gnecula committed 325 days ago

000b92f5

Fix file path regexp for Windows

gnecula committed 325 days ago

c1734f22

Update RNN kernels to use FFI.

dfm committed 325 days ago

c6e83903

Merge pull request #26380 from dfm:remove-legacy-custom-call-doc

Google-ML-Automation committed 325 days ago

cce3df10

Remove the tutorial about legacy custom calls.

dfm committed 325 days ago

21281f37

refactor: move array creation routines out of lax_numpy.py

jakevdp committed 325 days ago

7bacfbc6

Merge pull request #26375 from jakevdp:fix-doc

Google-ML-Automation committed 325 days ago

ce51384b

Merge pull request #26372 from jakevdp:split-lax-numpy

Google-ML-Automation committed 325 days ago

af29a7b0

DOC: update numpy/scipy versions in deprecation doc

jakevdp committed 325 days ago

e0ee4156

Newer Older