ROCm/jax

Pull Requests Commits

Ruturaj4 committed 1 year ago

52ab0501

Merge pull request #20629 from Sai-Suraj-27:prefer_TypeErorr

a-googler committed 1 year ago

4e889ce8

Merge pull request #20578 from carlosgmartin:logsumexp_where

a-googler committed 1 year ago

6ca46723

Merge pull request #20589 from jakevdp:beta-sign

a-googler committed 1 year ago

c45cdc9c

Merge pull request #20642 from jakevdp:actions-deps

a-googler committed 1 year ago

e186e2f3

Merge pull request #20179 from keshavb96:custom_gpu_ops_partitioning

a-googler committed 1 year ago

f380c3c3

CI: pin actions dependencies to most recent release

jakevdp committed 1 year ago

f0295321

BUG: fix sign of beta()

jakevdp committed 1 year ago

97b15cc6

Merge pull request #20528 from rajasekharporeddy:test_branch1

a-googler committed 1 year ago

66427f88

Add where argument to logsumexp.

carlosgmartin committed 1 year ago

e98612e2

Merge pull request #20622 from Sai-Suraj-27:updated_pre-commit

a-googler committed 1 year ago

1d26365e

Merge pull request #20623 from Sai-Suraj-27:update_ubuntu_version

a-googler committed 1 year ago

f6a46840

Merge pull request #20634 from hawkinsp:warn

a-googler committed 1 year ago

e107cee8

Fix test failure in complex clip test.

hawkinsp committed 1 year ago

c2ba8fcf

Current status + build script fixes

nouiz committed 1 year ago

83ffcc9c

Switch `llo` and `tpu` dialects to MLIR properties.

chsigg committed 1 year ago

5d540433

Prefer raising of TypeError for invalid types instead of ValueError.

Sai-Suraj-27 committed 1 year ago

55645213

Update XLA dependency to use revision

a-googler committed 1 year ago

f2fc72d6

Remove `mesh_shape` from ShardingSpec since it's not in use anymore.

yashk2810 committed 1 year ago

7fe14916

Delete deprecated AOT layouts API.

yashk2810 committed 1 year ago

c3f5af7d

Updated ubuntu os version in readthedocs.yml file to the latest suggested version.

Sai-Suraj-27 committed 1 year ago

8bbb8bc8

Updated all the pre-commit hooks versions.

Sai-Suraj-27 committed 1 year ago

29def4ee

Merge pull request #20380 from kaixih:cudnn_attention_dev

a-googler committed 1 year ago

9a931af6

Update XLA dependency to use revision

a-googler committed 1 year ago

062836ea

Remove deprecated code from JAX lowering and compilation

yashk2810 committed 1 year ago

c6804f92

Share lowering code between jit and aot jit path

yashk2810 committed 1 year ago

3b5980fd

Merge pull request #20266 from mattjj:earray

a-googler committed 1 year ago

e8b86cd8

start adding EArray, a jax.Array analog that can contain extended dtypes

mattjj committed 1 year ago

89f26db3

jax.pure_callback and jax.experimental.io_callback now use jax.Arrays

superbobry committed 1 year ago

9616900c

Update XLA dependency to use revision

a-googler committed 1 year ago

63aee947

Older