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.5.3 source code.tar.gz	2025-03-19	15.9 MB	0
JAX v0.5.3 source code.zip	2025-03-19	16.9 MB	0
README.md	2025-03-19	414 Bytes	0
Totals: 3 Items		32.9 MB	0

New Features
Added a allow_negative_indices option to jax.lax.dynamic_slice, jax.lax.dynamic_update_slice and related functions. The default is true, matching the current behavior. If set to false, JAX does not need to emit code clamping negative indices, which improves code size.
Added a replace option to jax.random.categorical to enable sampling without replacement.

Source: README.md, updated 2025-03-19

Other Useful Business Software

No-Nonsense Code-to-Cloud Security for Devs | Aikido

Connect your GitHub, GitLab, Bitbucket, or Azure DevOps account to start scanning your repos for free.

Aikido provides a unified security platform for developers, combining 12 powerful scans like SAST, DAST, and CSPM. AI-driven AutoFix and AutoTriage streamline vulnerability management, while runtime protection blocks attacks.

Start for Free

MongoDB Atlas | Run databases anywhere

Ensure the availability of your data with coverage across AWS, Azure, and GCP on MongoDB Atlas—the multi-cloud database for every enterprise.

MongoDB Atlas allows you to build and run modern applications across 125+ cloud regions, spanning AWS, Azure, and Google Cloud. Its multi-cloud clusters enable seamless data distribution and automated failover between cloud providers, ensuring high availability and flexibility without added complexity.

Learn More

Recommended Projects

AwesomeBump
Open Source graphic tool and alternative to Insane Bump
Clonezilla
A partition and disk imaging/cloning program
DeSmuME: Nintendo DS emulator
DeSmuME is a Nintendo DS emulator
7-Zip
A free file archiver for extremely high compression
Apache OpenOffice
The free and Open Source productivity suite

JAX Files

Composable transformations of Python+NumPy programs

Get an email when there's a new version of JAX