Home / Python / jax

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python ◇ jax Apache-2.0

View on GitHub Website

★36.3KSTARS

⑂3.8KFORKS

!2,601ISSUES

🏆#287GLOBAL RANK

🔥5DAYS TRENDING

🚀

Maintainer Growth Kit for jax

Claim this project, add your verified backlink badge to your README, and download milestone cards.

Claim Repo

Star History

Continuous Observations

Interactive star growth chart for jax-ml/jax

Momentum

+909

STARS · LAST 30 DAYS

PER DAY

#65

MOST-STARRED Python

Window	7 days	30 days	90 days
Stars gained	+210	+909	+2.7K
Per day	30	30	30
Forks gained	+19	+76	+191

jax gained 909 stars in the last 30 days, about 30 a day, and now has 36.3K. It is about 8 years old and has averaged roughly 4.5K stars a year. It ranks #65 among Python repositories and #287 across all languages on GitHubRepo.

Trending Record

jax has maintained a continuous presence across global trending indexes, peaking at #472. Below is the 30-day activity profile:

💡 Overview

jax is an open-source project written in Python: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

Engineered for speed, consistency, and developer ease, it solves common hurdles in jax. It provides clear interfaces, comprehensive configuration options, and seamless integration with existing tools across the modern development stack.

⚡ Key Features

Optimized execution pipeline written in Python for predictable speed.

Zero-friction configuration with comprehensive sensible defaults out of the box.

Cross-platform runtime support across Linux, macOS, and Windows environments.

Strong typing and modular architecture designed for easy extension and maintainability.

Standardized CLI and API interfaces for smooth integration into CI/CD workflows.

Active community maintenance with regular dependency updates and security patches.

📥 Installation

terminal

$ pip install jax

$ pip install jax # or: brew install jax

> pip install jax

⚙ System Requirements

Platforms

• macOS
• Linux
• Windows

Runtime & Dependencies

Python >= 3.9, pip, virtualenv

Architecture

x86_64, ARM64 (Apple Silicon & Graviton)

🧠 How It Works

jax coordinates its core functionality through a modular Python pipeline. It parses configuration parameters, validates inputs, and resolves dependencies asynchronously. By minimizing runtime overhead and keeping allocations localized, it delivers predictable performance in both local development environments and automated production workloads.

🎯 Production Use Cases

Production System Integration

Embed jax into Python backend services to handle core application logic.

CI/CD Automated Pipelines

Run automated validation, builds, and integration suites during deployments.

Developer Tooling & Workflows

Accelerate developer onboarding with pre-configured project utilities.

Open Source Extension

Fork and customize internal modules under the repository's open Apache-2.0 license.

🚀 Getting Started

Install jax using your package manager: `pip install jax`

Initialize your project workspace or configuration file for jax.

Import jax into your codebase or invoke it directly from your terminal.

Execute your test suite or run `jax --help` to verify successful setup.

👍 Strengths

Active community backing with 36,332 GitHub stars and verified adoption.

Permissive open-source distribution under the Apache-2.0 license.

Built in Python for high execution speed and developer familiarity.

Cross-platform compatibility across modern Linux, macOS, and Windows environments.

Clean modular design allowing flexible configuration and pipeline integration.

⚠️ Considerations

Requires familiarity with Python and modern CLI workflows.

Ecosystem extensions may require manual configuration depending on environment constraints.

Active development roadmap means breaking API changes may occur across major versions.

⇄ Alternatives & Direct Competitors

public-apis/public-apis ★ 483.8K Python

A collective list of free APIs

Compare ↗

EbookFoundation/free-programming-books ★ 398K Python

:books: Freely available programming books

Compare ↗

practical-tutorials/project-based-learning ★ 284.9K Python

Curated list of project-based tutorials

Compare ↗

NousResearch/hermes-agent ★ 249.1K Python

The agent that grows with you

Compare ↗

👥 Who Should Use This

Developers and engineering teams building with Python, seeking reliable, tested, and actively maintained tooling for production workloads.

🏆 Nearby in the Rankings

jax-ml/jax is currently ranked #287 by stars across every repository tracked on GitHubRepo. These are adjacent projects:

Rank	Repository	Language	Stars	Action
#282	firecracker-microvm/firecracker	Rust	★ 37K	Compare ↗
#283	nextcloud/server	PHP	★ 36.9K	Compare ↗
#284	sgl-project/sglang	Python	★ 36.5K	Compare ↗
#285	gitroomhq/postiz-app	TypeScript	★ 36.4K	Compare ↗
#286	wailsapp/wails	Go	★ 36.4K	Compare ↗
#287	jax-ml/jax This Project	Python	★ 36.3K
#288	permissionlesstech/bitchat	Swift	★ 36.3K	Compare ↗
#289	solidjs/solid	TypeScript	★ 36.1K	Compare ↗
#290	gchq/CyberChef	JavaScript	★ 36K	Compare ↗
#291	utmapp/UTM	Swift	★ 35.7K	Compare ↗
#292	zellij-org/zellij	Rust	★ 35.6K	Compare ↗

Frequently Asked Questions

What does jax do? +

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

What language is jax written in? +

The primary language is Python. Topics include: jax.

Is jax actively maintained? +

Yes, the last recorded push was on Sep 26, 2026 with 2,601 open issues being tracked.

How many stars does jax have? +

jax has 36,342 stars and 3,811 forks on GitHub.

How does jax rank among GitHub repositories? +

With 36,342 stars, jax-ml/jax is ranked #287 globally across all repositories tracked on GitHubRepo and #65 among Python projects.

What license is jax distributed under? +

The repository reports a Apache-2.0 license. Always verify the repository LICENSE file for legal terms.

👑 MAINTAINER VERIFICATION OPEN

Claim this repository

Are you the creator or maintainer of jax-ml/jax? Claiming your repository unlocks priority video reel generation, verified creator status, and milestone achievement announcements sent to your inbox.

Claim jax ↗ Free & instant verification

⭐ Featured on GitHubRepo.cloud Badge

Add our official badge to your repository's README.md. It gives your visitors proof of recognition, real-time star velocity tracking, and builds a permanent verified backlink.

GitHub Markdown (Paste into your README.md)

✓ Recommended

markdown

[![Featured on GitHubRepo.cloud](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge)

[![Featured on GitHubRepo.cloud](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge)

Full Badge Trio (Featured + Stars + Global Rank)

markdown

[![Featured on GitHubRepo.cloud](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge) [![GitHubRepo Stars](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=stars)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge) [![Global Rank](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=rank)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge)

[![Featured on GitHubRepo.cloud](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge) [![GitHubRepo Stars](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=stars)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge) [![Global Rank](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=rank)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge)

HTML (For websites & documentation)

html

<a href="https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge"><img src="https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured" alt="Featured on GitHubRepo.cloud" height="26"></a>

<a href="https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge"><img src="https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=featured" alt="Featured on GitHubRepo.cloud" height="26"></a>

Download SVG Badge Download PNG Card (1200×630) Share on X / Twitter ↗ Share on LinkedIn ↗

🏆 Auto-Generated Achievement Cards

Official milestone cards certified by GitHubRepo.cloud data. Download high-resolution visuals to share your community traction on LinkedIn, X, or your repository discussion boards.

Official achievement card for jax-ml/jax

Download PNG Card (1200×630) Download SVG Vector Post Milestone to X ↗ Post to LinkedIn ↗

Markdown Image for README / Blog

markdown

[![Top 1% GitHub Repository This Week on GitHubRepo.cloud](https://githubrepo.cloud/cards/jax-ml/jax/achievement.svg?type=top1)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=achievement_card)

[![Top 1% GitHub Repository This Week on GitHubRepo.cloud](https://githubrepo.cloud/cards/jax-ml/jax/achievement.svg?type=top1)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=achievement_card)

🌐 Multi-Platform Developer Community Drafts

High-converting, non-spammy promotion drafts crafted specifically for engineering communities (LinkedIn, Reddit, Hacker News, Dev.to, X, and Threads). Copy and customize for your project launch or updates.

💼

LinkedIn

Why jax is replacing paid subscriptions for engineering teams

Stop paying $50+/month for locked SaaS tools.

Over 36,342 developers have starred jax-ml/jax, an impressive Python open-source project designed to Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

Key Engineering Advantages:
• 100% Local-First: Complete data privacy with zero tracking or telemetry.
• Zero Subscription Costs: No paywalls, per-seat fees, or API markups.
• Complete Customization: Native open-source architecture you can fork, inspect, and deploy anywhere.
• Proven Adoption: 36,342 GitHub stars and 3,811 forks.

If you're looking to streamline your engineering workflows without ballooning software expenses, this is worth checking out.

👉 Full star velocity and architectural report: https://githubrepo.cloud/repo/jax-ml/jax
⭐ GitHub Repository: https://github.com/jax-ml/jax

#OpenSource #SoftwareEngineering #DeveloperTools #Tech #Programming #Python

🤖

Reddit (r/opensource, r/webdev, r/selfhosted)

Show r/opensource: jax – A free, self-hosted tool that lets you Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Hey everyone,

I wanted to share an interesting open-source project called **jax** (https://github.com/jax-ml/jax) that has been picking up heavy momentum lately (36,342 stars).

### What problem does it solve?
Many existing tools in this domain require expensive cloud plans or store your data on third-party servers. jax was built with a local-first philosophy, allowing developers to Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more without recurring subscription costs or privacy compromises.

### Tech Stack & Details:
- **Language:** Python
- **License:** Apache-2.0
- **GitHub:** [jax-ml/jax](https://github.com/jax-ml/jax)
- **Live Star Momentum Tracker:** [https://githubrepo.cloud/repo/jax-ml/jax](https://githubrepo.cloud/repo/jax-ml/jax)

Has anyone here tested this in production or integrated it into existing pipelines? Would love to hear your experience and feedback!

🟧

Hacker News (Show HN)

Show HN: jax – Composable transformations of Python+NumPy programs: differentiate,...

Hi HN,

jax is an open-source Python project designed to Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

Rather than relying on closed cloud endpoints, it executes locally on your own machine, providing predictable latency, total privacy, and full control.

Repository: https://github.com/jax-ml/jax
Documentation & Metrics: https://githubrepo.cloud/repo/jax-ml/jax

Feedback and PRs are welcome.

✍️

Dev.to / Hashnode Tags: opensource, python, devtools, webdev

This GitHub repo replaces a $50 a month tool! Stop paying expensive software subscriptions. Meet jax (36,342 GitHub Stars)

---
title: This GitHub repo replaces a $50 a month tool! Stop paying expensive software subscriptions. Meet jax
published: true
tags: opensource, python, devtools, programming
---

## Stop Paying Software Subscriptions For This

Finding high-quality, production-ready open source alternatives to expensive proprietary software can save engineering teams thousands of dollars every year.

Today, we're looking at **[jax-ml/jax](https://github.com/jax-ml/jax)**, a rapidly rising repository with **36,342 GitHub stars** and **3,811 forks**.

### What is jax?

> Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Written in **Python**, jax runs directly on your local workstation or private server, keeping your data secure and eliminating recurring license fees.

### Key Highlights
- **High Performance:** Lightweight, reliable, and built in Python.
- **Privacy & Security:** Complete ownership of your environment and data.
- **Growing Momentum:** Ranked **#287** globally across tracked repositories on [GitHubRepo](https://githubrepo.cloud/repo/jax-ml/jax).

### Quick Start
To get started locally:

```bash
git clone https://github.com/jax-ml/jax.git
cd jax
```

Check out the full repository and star history analytics here:
- **GitHub:** [jax-ml/jax](https://github.com/jax-ml/jax)
- **Star Velocity Tracker:** [jax on GitHubRepo.cloud](https://githubrepo.cloud/repo/jax-ml/jax)

𝕏

X / Twitter Communities

Thread / Community Post for jax

Developers are sleeping on this GitHub repo 👇

jax-ml/jax (36,342 ⭐) replaces expensive monthly software subscriptions with 100% free open source code.

⚡ What it does: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
💻 Built with: Python
🔒 100% local execution & privacy

Check live star velocity and benchmarks:
https://githubrepo.cloud/repo/jax-ml/jax

#opensource #developer #coding #tech #programming

🧵

Threads Topics

Topic Discussion: Open Source & Developer Tools

This GitHub repo replaces a $50/month tool! 🚀

If you want to save hours of manual work and stop paying subscriptions, check out jax-ml/jax.

It already has over 36,342 stars and lets you Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

Drop a comment if you've tried it, or check the full star momentum report:
https://githubrepo.cloud/repo/jax-ml/jax

From our network

jax-ml/jax

Star History

Momentum

Trending Record

💡 Overview

⚡ Key Features

📥 Installation

⚙ System Requirements

Platforms

Runtime & Dependencies

Architecture

🧠 How It Works

🎯 Production Use Cases

Production System Integration

CI/CD Automated Pipelines

Developer Tooling & Workflows

Open Source Extension

🚀 Getting Started

👍 Strengths

⚠️ Considerations

⇄ Alternatives & Direct Competitors

👥 Who Should Use This

🏆 Nearby in the Rankings

Frequently Asked Questions

Claim this repository

⭐ Featured on GitHubRepo.cloud Badge

GitHub Markdown (Paste into your README.md)

Full Badge Trio (Featured + Stars + Global Rank)

HTML (For websites & documentation)

🏆 Auto-Generated Achievement Cards

Markdown Image for README / Blog

🌐 Multi-Platform Developer Community Drafts

Built something? Put it in front of millions of developers.