JAX Toolbox Files

Public CI, Docker images for popular JAX libraries

This is an exact mirror of the JAX Toolbox project, hosted at https://github.com/NVIDIA/JAX-Toolbox. SourceForge is not affiliated with JAX Toolbox.

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
JAX-Toolbox source code.tar.gz	2026-05-14	18.2 MB	0
JAX-Toolbox source code.zip	2026-05-14	18.3 MB	3
Totals: 2 Items		36.5 MB	3

Other Useful Business Software

Stop Storing Third-Party Tokens in Your Database

Auth0 Token Vault handles secure token storage, exchange, and refresh for external providers so you don't have to build it yourself.

Rolling your own OAuth token storage can be a security liability. Token Vault securely stores access and refresh tokens from federated providers and handles exchange and renewal automatically. Connected accounts, refresh exchange, and privileged worker flows included.

Try Auth0 for Free

Full-stack observability with actually useful AI | Grafana Cloud

Our generous forever free tier includes the full platform, including the AI Assistant, for 3 users with 10k metrics, 50GB logs, and 50GB traces.

Built on open standards like Prometheus and OpenTelemetry, Grafana Cloud includes Kubernetes Monitoring, Application Observability, Incident Response, plus the AI-powered Grafana Assistant. Get started with our generous free tier today.

Create free account

Full-stack observability with actually useful AI | Grafana Cloud

Our generous forever free tier includes the full platform, including the AI Assistant, for 3 users with 10k metrics, 50GB logs, and 50GB traces.

Create free account

Recommended Projects

MaxText
A simple, performant and scalable Jax LLM
jax
Just an Extended Artificial Intelligence
Swagger Core
Examples, server integrations for generating Swagger API Specification
Brax
Massively parallel rigidbody physics simulation
TensorLy
Tensor Learning in Python

JAX Toolbox Files

Public CI, Docker images for popular JAX libraries

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