jax-ml/jax

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

GitHub repository with 35,759 stars and 3,609 forks.

Language: Python

Topics: jax

Open provider repository

24h trend summary

Trending score 1.23, activity score 0.05, stars gained +18, forks gained +1.

Latest metric snapshot

2026-06-05: 35,759 stars and 3,609 forks.

Similar repositories

  1. 1. jax-ml/jax

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

    GitHub repository with 35,759 stars and 3,609 forks.

    Trending score: 1.23; stars gained: +18; forks gained: +1.

    Language: Python

    Topics: jax

  2. 2. AI-Hypercomputer/maxtext

    A simple, performant and scalable Jax LLM!

    GitHub repository with 2,309 stars and 527 forks.

    Trending score: 0.98; stars gained: +1; forks gained: +0.

    Language: Python

    Topics: large-language-models, llm, gpt, deepseek, fine-tuning, gemma2

  3. 3. keras-team/keras

    Deep Learning for humans

    GitHub repository with 64,088 stars and 19,747 forks.

    Trending score: 0.65; stars gained: +3; forks gained: -1.

    Language: Python

    Topics: deep-learning, tensorflow, neural-networks, machine-learning, data-science, python

  4. 4. PennyLaneAI/pennylane

    PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

    GitHub repository with 3,247 stars and 809 forks.

    Trending score: 0.60; stars gained: +3; forks gained: +14.

    Language: Python

    Topics: quantum, machine-learning, deep-learning, neural-network, optimization, quantum-computing

  5. 5. PASSIONLab/OpenEquivariance

    OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

    GitHub repository with 149 stars and 9 forks.

    Trending score: 0.45; stars gained: +1; forks gained: +0.

    Language: Python

    Topics: cuda, geometric-deep-learning, graph-neural-networks, sparse-tensors, equivariance, hip

  6. 6. team-ocean/veros

    The versatile ocean simulator, in pure Python, powered by JAX.

    GitHub repository with 395 stars and 59 forks.

    Trending score: 0.19; stars gained: +0; forks gained: +0.

    Language: Python

    Topics: oceanography, python, parallel, multi-core, geophysics, jax

Trending in Python

  1. 1. NousResearch/hermes-agent

    The agent that grows with you

    GitHub repository with 182,335 stars and 31,270 forks.

    Trending score: 5.95; stars gained: +1,867; forks gained: +361.

    Language: Python

    Topics: ai, ai-agent, ai-agents, anthropic, chatgpt, claude

  2. 2. chopratejas/headroom

    Compress tool outputs, logs, files, and RAG chunks before they reach the LLM. 60-95% fewer tokens, same answers. Library, proxy, MCP server.

    GitHub repository with 14,053 stars and 885 forks.

    Trending score: 5.69; stars gained: +2,829; forks gained: +175.

    Language: Python

    Topics: agent, ai, anthropic, compression, context-engineering, context-window

  3. 3. Imbad0202/academic-research-skills

    Academic Research Skills for Claude Code: research → write → review → revise → finalize

    GitHub repository with 27,548 stars and 2,267 forks.

    Trending score: 5.52; stars gained: +1,079; forks gained: +89.

    Language: Python

    Topics: academic-pipeline, academic-writing, ai-research, claude, claude-code, literature-review

  4. 4. rohitg00/ai-engineering-from-scratch

    Learn it. Build it. Ship it for others.

    GitHub repository with 28,711 stars and 4,695 forks.

    Trending score: 5.32; stars gained: +1,261; forks gained: +238.

    Language: Python

    Topics: agents, ai, ai-agents, ai-engineering, computer-vision, course

  5. 5. vinta/awesome-python

    An opinionated list of Python frameworks, libraries, tools, and resources

    GitHub repository with 301,427 stars and 28,046 forks.

    Trending score: 4.60; stars gained: +518; forks gained: +24.

    Language: Python

    Topics: awesome, python, collections, python-frameworks, python-libraries, python-tools

  6. 6. langchain-ai/langchain

    The agent engineering platform.

    GitHub repository with 138,587 stars and 22,961 forks.

    Trending score: 4.53; stars gained: +171; forks gained: +31.

    Language: Python

    Topics: agents, ai, ai-agents, anthropic, chatgpt, deepagents

Trending topic: jax

  1. 1. jax-ml/jax

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

    GitHub repository with 35,759 stars and 3,609 forks.

    Trending score: 1.23; stars gained: +18; forks gained: +1.

    Language: Python

    Topics: jax

  2. 2. AI-Hypercomputer/maxtext

    A simple, performant and scalable Jax LLM!

    GitHub repository with 2,309 stars and 527 forks.

    Trending score: 0.98; stars gained: +1; forks gained: +0.

    Language: Python

    Topics: large-language-models, llm, gpt, deepseek, fine-tuning, gemma2

  3. 3. keras-team/keras

    Deep Learning for humans

    GitHub repository with 64,088 stars and 19,747 forks.

    Trending score: 0.65; stars gained: +3; forks gained: -1.

    Language: Python

    Topics: deep-learning, tensorflow, neural-networks, machine-learning, data-science, python

  4. 4. PennyLaneAI/pennylane

    PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

    GitHub repository with 3,247 stars and 809 forks.

    Trending score: 0.60; stars gained: +3; forks gained: +14.

    Language: Python

    Topics: quantum, machine-learning, deep-learning, neural-network, optimization, quantum-computing

  5. 5. PASSIONLab/OpenEquivariance

    OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

    GitHub repository with 149 stars and 9 forks.

    Trending score: 0.45; stars gained: +1; forks gained: +0.

    Language: Python

    Topics: cuda, geometric-deep-learning, graph-neural-networks, sparse-tensors, equivariance, hip

  6. 6. team-ocean/veros

    The versatile ocean simulator, in pure Python, powered by JAX.

    GitHub repository with 395 stars and 59 forks.

    Trending score: 0.19; stars gained: +0; forks gained: +0.

    Language: Python

    Topics: oceanography, python, parallel, multi-core, geophysics, jax