jax

jax-ml/jax

+33

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

kaggle-tpu-lab

+40

Frontier-class open models on a free Kaggle TPU v5e-8: GLM-5.3-Flash 320B MoE (~165 tok/s, our own JAX engine) and Qwen3.8-27B bf16 (~130 tok/s), 262k context, prefix caching. Works with Claude Code, Codex, opencode and pi.

keras

+17

Deep Learning for humans

trainer

+12

Distributed AI Model Training and LLM Fine-Tuning on Kubernetes

scaling-book

+15

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

google/flax

+10

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

TransformerEngine

+9

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

EMI-Group/evox

+9

Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.

iree-org/iree

+9

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