Home » Oreps » Jax » All Releases » jaxlib-v0.4.19 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Directory Browser
Velocity History
Releasesjaxlib-v0.4.19
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jaxlib-v0.4.19

Jax: jaxlib-v0.4.19 Release

Release date:
October 18, 2023
Previous version:
jaxlib-v0.4.18 (released October 6, 2023)
Magnitude:
7,111 Diff Delta
Contributors:
24 total committers
Data confidence:
Commits:
114 impacting commits

114 Commits in this Release

Ordered by the degree to which they evolved the repo in this version.

MAINT Migrate remaining internal/test modules to use state objects
Authored October 12, 2023
[pallas] Simplify several primitives bindings
Authored October 10, 2023
Migrate another subset of internal modules to use state objects
Authored October 11, 2023
[export] Fix the call_exported in presence of shardings
Authored October 6, 2023
[pallas:gpu] De-duplicate lowering code for JAX primitives that map to Triton builtins
Authored October 13, 2023
Use totalorder comparisons for sort
Authored October 13, 2023
Add segment_ids support to pallas flash attention on TPU
Authored October 10, 2023
[export] Adapt several collective lowering rules for multi-platform lowering
Authored October 6, 2023
[Mosaic] apply_vector_layout C++ rewrite (18): vector.transpose
Authored October 9, 2023
[Mosaic] apply_vector_layout C++ rewrite: Handle elementwise ops by checking for the Elementwise trait and using the generic Operation interface, without templates
Authored October 9, 2023
[Pallas TPU] Add DMA descriptor abstraction for constructing but not starting DMAs
Authored October 17, 2023
[jax2tf] First step to enable multi-platform native lowering
Authored October 13, 2023
[pallas:gpu] Implement `get` and `swap` using `load` and `masked_swap` lowering rules
Authored October 13, 2023
[pallas:gpu] Simplify reshape lowering rule
Authored October 16, 2023
Optimize make_array_from_callback for fully replicated shardings by going via batched_device_put
Authored October 11, 2023
MAINT Use a generator expression with `all()` and `any()`
Authored October 10, 2023
[pallas] `dot` fixes
Authored October 16, 2023
[pallas:gpu] Simplify `broadcast_to`, `min`, `max` lowering
Authored October 17, 2023
[Mosaic] apply_vector_layout C++ rewrite: add tpu.concatenate
Authored October 17, 2023
random_test: fix deprecation warnings for key tests
Authored October 6, 2023
[random] Avoid references to PRNGKeyArray type
Authored October 13, 2023
jnp.frompyfunc: fix .at() edge case
Authored October 9, 2023
jax.core: deprecate some inadvertent exports
Authored October 11, 2023
Fix mypy errors in flash_attention
Authored October 10, 2023
Make sure we don't filter stack frames of packages that start with a `jax` prefix
Authored October 17, 2023
Remove calling configure_library_path during jax import and get libtpu path from libtpu_module.get_library_path()
Authored October 10, 2023
feat: add wrapcauchy logpdf and pdf
Authored October 16, 2023
Set up an API to top trace and fdo profile in memory
Authored October 13, 2023
typing: add some type assertions to typing_test
Authored October 13, 2023
jax.make_jaxpr: fix __name__ & related attributes
Authored October 9, 2023
Remove take_ownership support in DLPack
Authored October 10, 2023
Set RPATH, not RUNPATH in JAX CUDA builds
Authored October 10, 2023
[pallas:gpu] Directly use Triton built-in function lowering for `max_contiguous` and `multiple_of`
Authored October 16, 2023
Cleanup lowering rule for hlo_unshard, to remove platform dependence
Authored October 12, 2023
Delete cuda 12.0.1 rbe configs since JAX doesn't support it anymore
Authored October 13, 2023
Only perform compilation cache writes from process 0
Authored October 9, 2023
[XLA:GPU] Fix bug in all-to-all for complex data types
Authored October 16, 2023
[PJRT C API] Set the gpu plugin allocator related options
Authored October 13, 2023
Rollback to fix internal breakage
Authored October 17, 2023
Make jax._src.typing.DTypeLike more strictly defined
Authored October 10, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.23 Released December 14, 2023
704 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
6,291 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
18,528 Ξ”
jaxlib-v0.4.20 Released November 2, 2023
4,475 Ξ”
jaxlib-v0.4.19 Released October 18, 2023
7,111 Ξ”
jaxlib-v0.4.18 Released October 6, 2023
4,220 Ξ”
jaxlib-v0.4.17 Released October 3, 2023
9,022 Ξ”
jaxlib-v0.4.16 Released September 18, 2023
9,933 Ξ”
jaxlib-v0.4.15 Released August 30, 2023
15,453 Ξ”
jaxlib-v0.4.14 Released July 27, 2023
17,116 Ξ”

Top Contributors in jaxlib-v0.4.19

superbobry
chr1sj0nes
gnecula
jakevdp
tlongeri
a-googler
majnemer
yashk2810
jyingl3
sharadmv

Directory Browser for jaxlib-v0.4.19

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy