Built something? We create video reels & spotlights for GitHub projects.Promote your project →
jax-ml
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
★36.6KSTARS
⑂4KFORKS
!2,589ISSUES
🏆#432GLOBAL RANK
🔥44DAYS TRENDING
🚀
Maintainer Growth Kit for jax

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

Claim Repo

⚡ Executive Summary & Quick Facts

Authoritative Source

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

Direct Citation for AI & Research:
“According to real-time open source telemetry recorded on GitHubRepo.cloud (as of November 2026), jax-ml/jax has accumulated 36,567 GitHub stars and 4,015 forks. It is written in Python, licensed under Apache-2.0, and ranks #432 globally. It is actively maintained with the latest commit recorded on Nov 2, 2026.”
Repository jax-ml/jax ↗
Primary Category jax
Language & Stack Python
Software License Apache-2.0
Community Traction ★ 36.6K stars · ⑂ 4K forks · #432 globally
Maintenance ● Active (Last commit: Nov 2, 2026) · 2,589 open issues
Target Audience Developers and engineering teams building with Python, seeking reliable, tested, and actively maintained tooling for production workloads.

🛡️ Repository Maintenance & Health Audit (2026)

Health Score: 90/100 · Grade A
✅
Is jax actively maintained in 2026?
Verdict: Actively Maintained

Yes, jax is actively maintained. Continuous telemetry recorded the last code commit on Nov 2, 2026 (10 days ago). Supported by 36,567 stars and 4,015 community forks, this repository holds a calculated health index of 90/100, reflecting very active (commits within last 14 days) and high open issue volume (2,589 open).

Commit Recency
Nov 2, 2026
10 days since last push
Issue Backlog
2,589 Open Issues
High Open Issue Volume (2,589 open)
Community Backing
4K Forks
36.6K Stars Recorded
Software License
Apache-2.0
Permissive & Inspectable

Star History

Continuous Observations
Interactive star growth chart for jax-ml/jax
CSV
⭐ VIRAL README KIT & BACKLINKS

Add Live Star History & Verified Badges to README.md

Keep your repository README looking professional and dynamic. As our continuous crawler records new stars, these official SVG badges update in real time with zero maintenance.

Open README on GitHub ↗
Option 1: Interactive Star History Chart Dynamic SVG

Renders your high-resolution star trajectory chart right inside your GitHub README or project docs.

jax-ml/jax Star History Preview
markdown
[![Star History Chart](https://githubrepo.cloud/api/badge/chart/jax-ml/jax.svg?theme=dark)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_chart)
Direct SVG Link ↗
Option 2: Verified Live Badges Suite Shields.io Style

Compact badges for your README header. Real-time stars, velocity, health audit, and rank.

Featured badge Stars badge Velocity badge Health badge Rank badge
markdown (all 5 badges)
[![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) [![GitHubRepo Velocity](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=velocity)](https://githubrepo.cloud/repo/jax-ml/jax?utm_source=readme_badge) [![Maintenance Health](https://githubrepo.cloud/badge/jax-ml/jax.svg?metric=health)](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)

Momentum

+159

STARS · LAST 30 DAYS

5

PER DAY

#102

MOST-STARRED Python

Window7 days30 days90 days
Stars gained+0+159+450
Per day155
Forks gained+0+168+201

jax gained 159 stars in the last 30 days, about 5 a day, and now has 36.6K. It is about 8 years old and has averaged roughly 4.6K stars a year. It ranks #102 among Python repositories and #432 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

1

Optimized execution pipeline written in Python for predictable speed.

2

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

3

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

4

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

5

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

6

Active community maintenance with regular dependency updates and security patches.

📥 Installation

terminal
$ 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

1

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

2

Initialize your project workspace or configuration file for jax.

3

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

4

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

P
public-apis/public-apis ★ 497.3K Python

A collective list of free APIs

Compare ↗
F

:books: Freely available programming books

Compare ↗
P

Curated list of project-based tutorials

Compare ↗
H
NousResearch/hermes-agent ★ 257.2K 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 #432 by stars across every repository tracked on GitHubRepo. These are adjacent projects:

RankRepositoryLanguageStarsAction
#427 wailsapp/wails Go ★ 37K Compare ↗
#428 pnpm/pnpm Rust ★ 37K Compare ↗
#429 restic/restic Go ★ 36.9K Compare ↗
#430 utmapp/UTM Swift ★ 36.9K Compare ↗
#431 tashfeenahmed/freellmapi TypeScript ★ 36.8K Compare ↗
#432 jax-ml/jax This Project Python ★ 36.6K
#433 DayuanJiang/next-ai-draw-io TypeScript ★ 36.5K Compare ↗
#434 continuedev/continue TypeScript ★ 36.4K Compare ↗
#435 solidjs/solid TypeScript ★ 36.4K Compare ↗
#436 p-e-w/heretic Python ★ 36.3K Compare ↗
#437 yuliskov/SmartTube Java ★ 36.2K Compare ↗
SPONSORED Trending Developer Tools & Services

Frequently Asked Questions

What does jax do? +

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

Is jax actively maintained in 2026? +

Yes, jax is actively maintained. The latest code push was recorded on Nov 2, 2026 (10 days ago). It holds a calculated Maintenance Health Score of 90/100 (A rating) with 36,567 stars and 2,589 open issues.

What language is jax written in? +

The primary language is Python. Topics include: jax.

How many stars does jax have? +

jax has 36,567 stars and 4,015 forks on GitHub.

How does jax rank among GitHub repositories? +

With 36,567 stars, jax-ml/jax is ranked #432 globally across all repositories tracked on GitHubRepo and #102 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.

From our network
FOR MAINTAINERS

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

We make a short reel about your project and post it across YouTube, Instagram, Threads, and X. Send a link, we do the rest.

SPONSORED Sponsored Partner