Home » Oreps » Jax » All Releases » jax-v0.4.19-rc 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
Releasesjax-v0.4.19-rc
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.4.19-rc

Jax: jax-v0.4.19-rc Release

Release date:
October 18, 2023
Previous version:
jax-v0.4.18 (released October 6, 2023)
Magnitude:
5,722 Diff Delta
Contributors:
24 total committers
Data confidence:
Commits:
112 impacting commits

112 Commits in this Release

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

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
[export] Adapt several collective lowering rules for multi-platform lowering
Authored October 6, 2023
Use totalorder comparisons for sort
Authored October 13, 2023
Add segment_ids support to pallas flash attention on TPU.
Authored October 10, 2023
[jax2tf] First step to enable multi-platform native lowering
Authored October 13, 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
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:gpu] Implement `get` and `swap` using `load` and `masked_swap` lowering rules.
Authored October 13, 2023
random_test: fix deprecation warnings for key tests
Authored October 6, 2023
[pallas:gpu] Simplify reshape lowering rule.
Authored October 16, 2023
[pallas] `dot` fixes.
Authored October 16, 2023
Set up an API to top trace and fdo profile in memory.
Authored October 13, 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] Avoid references to PRNGKeyArray type
Authored October 13, 2023
[XLA:GPU] Fix bug in all-to-all for complex data types.
Authored October 16, 2023
feat: add wrapcauchy logpdf and pdf
Authored October 16, 2023
typing: add some type assertions to typing_test
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
jax.make_jaxpr: fix __name__ & related attributes
Authored October 9, 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
Rollback to fix internal breakage
Authored October 17, 2023
Remove take_ownership support in DLPack.
Authored October 10, 2023
Set RPATH, not RUNPATH in JAX CUDA builds.
Authored October 10, 2023
MAINT Drop underscore from the name of externally-referenced state objects
Authored October 13, 2023
Only perform compilation cache writes from process 0.
Authored October 9, 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
Return PositionalSharding if input's rank is >= 3 or a NamedSharding if a mesh is available via the context from inspect_array_sharding. Never return GSPMDSharding from inspect_array_sharding.
Authored October 12, 2023
[PJRT C API] Set the gpu plugin allocator related options.
Authored October 13, 2023
[pallas] Add Triton lowering rule for `custom_jvp_call_p`.
Authored October 12, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.21 Released December 4, 2023
0 Ξ”
jax-v0.4.21-rc Released December 4, 2023
15,642 Ξ”
jax-v0.4.20 Released November 2, 2023
0 Ξ”
jax-v0.4.20-rc Released November 2, 2023
5,123 Ξ”
jax-v0.4.19 Released October 18, 2023
0 Ξ”
jax-v0.4.19-rc Released October 18, 2023
5,722 Ξ”
jax-v0.4.18 Released October 6, 2023
0 Ξ”
jax-v0.4.18-rc Released October 6, 2023
3,149 Ξ”
jax-v0.4.17 Released October 3, 2023
0 Ξ”
jax-v0.4.17-rc Released October 3, 2023
8,498 Ξ”

Top Contributors in jax-v0.4.19-rc

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

Directory Browser for jax-v0.4.19-rc

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