GV

GeezerVison

The World Wide Web, restored to 800 pixels of glory.

Home|About|Gaming cabinet|Archives|How it works|Starter sites

On this site

  • Sign in
  • ActionsCI/CD and automation
  • CodespacesCloud development environments
  • All Features
  • Documentation
  • Training
  • Certifications
  • Build with Copilot
  • Customize Copilot
  • Integrations
  • Trending repos
  • GitHub Sponsors
  • Developer articles
  • Enterprises

This view

  • Original page
  • New address

GeezerVison is showing github.com — Build software better, together

#

jax

Star

Here are 2,545 public repositories matching this topic...

Language: All
Filter by language
All 2,545 Python 1,908 Jupyter Notebook 426 C++ 26 HTML 24 Rust 20 Shell 13 TypeScript 13 TeX 9 JavaScript 8 C 4
Sort: Most stars
Sort options
Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

keras-team / keras

Star 64.4k

Deep Learning for humans

python data-science machine-learning deep-learning tensorflow pytorch neural-networks jax
  • Updated Nov 12, 2026
  • Python
jax

jax-ml / jax

Star 36.6k

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

jax
  • Updated Nov 12, 2026
  • Python

d2l-ai / d2l-en

Star 30k

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

python data-science machine-learning natural-language-processing reinforcement-learning computer-vision deep-learning mxnet book notebook tensorflow keras pytorch kaggle hyperparameter-optimization recommender-system gaussian-processes jax
  • Updated Aug 18, 2024
  • Python

lukasmasuch / best-of-ml-python

Star 23.9k

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

python nlp data-science machine-learning deep-learning tensorflow scikit-learn keras ml data-visualization pytorch transformer data-analysis gpt automl jax data-visualizations gpt-3 chatgpt
  • Updated Nov 5, 2026

kyegomez / OpenMythos

Sponsor
Star 14.9k

A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.

ai ml torch pytorch attention deepmind claude sonnet jax gpt-5 anthropic claude-ai claude-sonnet looped-transformers claude-code claude-code-plugin claude-mythos gpt-7
  • Updated May 23, 2026
  • Python
ivy

unifyai / ivy

Star 14.2k

Convert Machine Learning Code Between Frameworks

python tensorflow numpy pytorch jax
  • Updated Nov 10, 2026
  • Python
wandb

wandb / wandb

Star 11.3k

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

data-science machine-learning reinforcement-learning ai deep-learning tensorflow keras collaboration pytorch hyperparameter-optimization reproducibility hyperparameter-tuning data-versioning hyperparameter-search model-versioning jax mlops experiment-track ml-platform
  • Updated Nov 11, 2026
  • Python
einops

arogozhnikov / einops

Star 9.6k

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

deep-learning tensorflow numpy pytorch tensor mlx cupy jax einops
  • Updated Aug 26, 2026
  • Python

google / trax

Star 8.3k

Trax — Deep Learning with Clear Code and Speed

machine-learning reinforcement-learning deep-learning numpy deep-reinforcement-learning transformer jax
  • Updated Sep 26, 2025
  • Python

HenryNdubuaku / maths-cs-ai-compendium

Sponsor
Star 7.7k

Become a cracked AI/ML researcher/engineer with this unconventional textbook covering maths, computing, and ML with intuition.

python nlp computer-science machine-learning statistics reinforcement-learning computer-vision deep-learning math algorithms linear-algebra machine-learning-algorithms probability mathematics artificial-intelligence speech-processing multimodal-learning jax ai-textbook
  • Updated Jul 18, 2026
  • TypeScript

HIPS / autograd

Star 7.6k

Efficiently computes derivatives of NumPy code.

python machine-learning deep-learning neural-network numpy automatic-differentiation autograd scipy derivative backpropagation differentiation numpy-arrays jax ufunc
  • Updated Nov 9, 2026
  • Python

google / flax

Star 7.4k

Flax is a neural network library for JAX that is designed for flexibility.

jax
  • Updated Nov 11, 2026
  • Jupyter Notebook

probml / pyprobml

Star 7.2k

Python code for "Probabilistic Machine learning" book by Kevin Murphy

machine-learning tensorflow pytorch colab pml probabilistic-programming flax jupyter-notebooks pymc3 pyro jax numpyro blackjax
  • Updated Oct 27, 2026
  • Jupyter Notebook

BoltzmannEntropy / interviews.ai

Star 4.9k

It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.

python data-science machine-learning deep-learning information-theory jobs pytorch autograd artificial-intelligence feature-extraction ensemble-learning logistic-regression convolutional-neural-networks bayesian-statistics loss-functions interview-preparation pytorch-tutorial graduate-school jax
  • Updated Aug 22, 2025

sanchit-gandhi / whisper-jax

Star 4.7k

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

deep-learning speech-recognition speech-to-text whisper jax
  • Updated Apr 3, 2024
  • Jupyter Notebook

tensorflow / datasets

Star 4.6k

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

data machine-learning tensorflow numpy dataset datasets jax
  • Updated Nov 11, 2026
  • Python

lululxvi / deepxde

Star 4.5k

A library for scientific machine learning and physics-informed learning

deep-learning neural-network tensorflow pytorch operator pde paddle pinn jax scientific-machine-learning multi-fidelity-data physics-informed-learning deeponet
  • Updated Nov 8, 2026
  • Python

iree-org / iree

Star 4k

A retargetable MLIR-based machine learning compiler and runtime toolkit.

machine-learning compiler runtime tensorflow vulkan cuda pytorch spirv rocm onnx jax mlir
  • Updated Nov 12, 2026
  • C++

danijar / dreamerv3

Star 4k

Mastering Diverse Domains through World Models

minecraft reinforcement-learning artificial-intelligence general jax world-models
  • Updated Oct 4, 2026
  • Python

google-research / scenic

Star 3.8k

Scenic: A Jax Library for Computer Vision Research and Beyond

research computer-vision deep-learning transformers attention jax vision-transformer
  • Updated Oct 31, 2026
  • Python

Add this topic to your repo

To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."

Learn more

You are visitor 01402503 — GeezerVison does not store the pages you look up.
Best viewed with Netscape Navigator 4.7 or Internet Explorer 6 at 800×600. No cookies required.