ROCm/jax

Pull Requests Commits

Ruturaj4 committed 2 years ago

52ab0501

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

a-googler committed 2 years ago

4e889ce8

Merge pull request #20578 from carlosgmartin:logsumexp_where

a-googler committed 2 years ago

6ca46723

Merge pull request #20589 from jakevdp:beta-sign

a-googler committed 2 years ago

c45cdc9c

Merge pull request #20642 from jakevdp:actions-deps

a-googler committed 2 years ago

e186e2f3

Merge pull request #20179 from keshavb96:custom_gpu_ops_partitioning

a-googler committed 2 years ago

f380c3c3

CI: pin actions dependencies to most recent release

jakevdp committed 2 years ago

f0295321

BUG: fix sign of beta()

jakevdp committed 2 years ago

97b15cc6

Merge pull request #20528 from rajasekharporeddy:test_branch1

a-googler committed 2 years ago

66427f88

Add where argument to logsumexp.

carlosgmartin committed 2 years ago

e98612e2

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

a-googler committed 2 years ago

1d26365e

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

a-googler committed 2 years ago

f6a46840

Merge pull request #20634 from hawkinsp:warn

a-googler committed 2 years ago

e107cee8

Fix test failure in complex clip test.

hawkinsp committed 2 years ago

c2ba8fcf

Current status + build script fixes

nouiz committed 2 years ago

83ffcc9c

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

chsigg committed 2 years ago

5d540433

Prefer raising of TypeError for invalid types instead of ValueError.

Sai-Suraj-27 committed 2 years ago

55645213

Update XLA dependency to use revision

a-googler committed 2 years ago

f2fc72d6

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

yashk2810 committed 2 years ago

7fe14916

Delete deprecated AOT layouts API.

yashk2810 committed 2 years ago

c3f5af7d

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

Sai-Suraj-27 committed 2 years ago

8bbb8bc8

Updated all the pre-commit hooks versions.

Sai-Suraj-27 committed 2 years ago

29def4ee

Merge pull request #20380 from kaixih:cudnn_attention_dev

a-googler committed 2 years ago

9a931af6

Update XLA dependency to use revision

a-googler committed 2 years ago

062836ea

Remove deprecated code from JAX lowering and compilation

yashk2810 committed 2 years ago

c6804f92

Share lowering code between jit and aot jit path

yashk2810 committed 2 years ago

3b5980fd

Merge pull request #20266 from mattjj:earray

a-googler committed 2 years ago

e8b86cd8

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

mattjj committed 2 years ago

89f26db3

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

superbobry committed 2 years ago

9616900c

Update XLA dependency to use revision

a-googler committed 2 years ago

63aee947

Older