JAX Files

Composable transformations of Python+NumPy programs

This is an exact mirror of the JAX project, hosted at https://github.com/google/jax. SourceForge is not affiliated with JAX. For more information, see the SourceForge Open Source Mirror Directory.

The interactive file manager requires Javascript. Please enable it or use sftp or scp.
You may still browse the files here.

Name	Modified	Size	InfoDownloads / Week
Parent folder
JAX v0.6.1 source code.tar.gz	2025-05-21	17.1 MB	0
JAX v0.6.1 source code.zip	2025-05-21	18.2 MB	0
README.md	2025-05-21	821 Bytes	0
Totals: 3 Items		35.3 MB	0

New features:
Added jax.lax.axis_size which returns the size of the mapped axis given its name.
Changes
Additional checking for the versions of CUDA package dependencies was reenabled, having been accidentally disabled in a previous release.
JAX nightly packages are now published to artifact registry. To install these packages, see the JAX installation guide.
jax.sharding.PartitionSpec no longer inherits from a tuple.
jax.ShapeDtypeStruct is immutable now. Please use .update method to update your ShapeDtypeStruct instead of doing in-place updates.
Deprecations
jax.custom_derivatives.custom_jvp_call_jaxpr_p is deprecated, and will be removed in JAX v0.7.0.

Source: README.md, updated 2025-05-21

Other Useful Business Software

Our Free Plans just got better! | Auth0 Icon

Our Free Plans just got better! | Auth0

With up to 25k MAUs and unlimited Okta connections, our Free Plan lets you focus on what you do best—building great apps.

You asked, we delivered! Auth0 is excited to expand our Free and Paid plans to include more options so you can focus on building, deploying, and scaling applications without having to worry about your security. Auth0 now, thank yourself later.

Try free now

Simplify IT and security with a single endpoint management platform Icon

Simplify IT and security with a single endpoint management platform

Automate the hardest parts of IT

NinjaOne automates the hardest parts of IT, delivering visibility, security, and control over all endpoints for more than 20,000 customers. The NinjaOne automated endpoint management platform is proven to increase productivity, reduce security risk, and lower costs for IT teams and managed service providers. The company seamlessly integrates with a wide range of IT and security technologies. NinjaOne is obsessed with customer success and provides free and unlimited onboarding, training, and support.

Learn More

Recommended Projects

AwesomeBump
Open Source graphic tool and alternative to Insane Bump
DeSmuME: Nintendo DS emulator
DeSmuME is a Nintendo DS emulator
7-Zip
A free file archiver for extremely high compression
KeePass
A lightweight and easy-to-use password manager
Clonezilla
A partition and disk imaging/cloning program