8000 Commits · cky9301/jax · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on Oct 5, 2022

Handle FP8 types.

Artem-B
authored and
jax authors
committed
Add `addressable_shards` to SDA and DA as a compatibility API to match with `jax.Array`. This will aid in transition to `jax.Array`.

yashk2810
authored and
jax authors
committed
Merge pull request #12654 from jakevdp:fft-typing
jax authors
committed
Merge pull request #12660 from hawkinsp:testing
jax authors
committed
Add sharding to `DeviceArray` and `ShardedDeviceArray` as a compatibility change to rollout `jax.Array`.

yashk2810
authored and
jax authors
committed
Add `host_local_array_to_global_array` and `global_array_to_host_local_array` for enabling transition to jax.Array.

yashk2810
authored and
jax authors
committed
Migrate api_test, lax_numpy_test, and lax_vmap_test to
hawkinsp
committed
Merge pull request #12657 from jakevdp:linalg-typing
jax authors
committed

Commits on Oct 4, 2022

[typing] add annotations to numpy.fft
jakevdp
committed
[typing] add annotations 10000 to jax.numpy.linalg
jakevdp
committed
Merge pull request #12658 from sharadmv:fix-profiler
jax authors
committed
Fix collect_profile _src import
sharadmv
committed
Merge pull request #12640 from mattjj:djax-slice-sick
jax authors
committed
[dynamic-shapes] small fix to einsum (and indexing)
mattjj
committed
Test that array layout is preserved in Python callbacks

sharadmv
authored and
jax authors
committed
Merge pull request #12639 from skye:version
jax authors
committed
Merge pull request #12641 from jakevdp:asarray-type
jax authors
committed
Merge pull request #12653 from jakevdp:jacobian-wraps
jax authors
committed
jax.jacobian: propagate function signature to transformed function
jakevdp
committed
Add types to jax/_src/numpy/util.py
jakevdp
committed
[sparse] Add conversions between BCSR and BCOO.

tlu7
authored and
jax authors
committed
Create `Array`s from `__getitem__` and `__iter__`. This is done by `device_put`ting from the host to default device which is suboptimal. But there is a TODO to fix this!

yashk2810
authored and
jax authors
committed
Merge pull request #12628 from hawkinsp:testing
jax authors
committed
Add an internal jtu.sample_product test decorator.
hawkinsp
committed

Commits on Oct 3, 2022

Merge pull request #12637 from jakevdp:setops-typing
jax authors
committed
Update version and CHANGELOG for jax 0.3.21 release
skye
committed
Merge pull request #12591 from sudhakarsingh27:add_pytest_run_for_jaxlib_release
jax authors
committed
Merge pull request #12612 from ROCmSoftwarePlatform:rocm_53_enhancements
jax authors
committed
Merge pull request #12630 from google:dependabot/github_actions/styfle/cancel-workflow-action-0.10.1
jax authors
committed
[typing] add full annotations for lax_numpy setops
jakevdp
committed
`get_device_buffers()` on ShardedBuffer if config.jax_array is enabled because jax.Arrray does not work with ShardedBuffer since jax.Array is like a ShardedBuffer.

yashk2810
authored and
jax authors
committed
Merge pull request #12624 from sergedurand:fix-books-link
jax authors
committed
Merge pull request #12611 from mattjj:custom-vjp-improve-type-error-checking
jax authors
committed
[sparse] Move broadcasting_vmap to sparse util.

tlu7
authored and
jax authors
committed
Add unsafe_buffer_pointer to _DeviceArray

krishnaharidasan
authored and
jax authors
committed

0