Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

🐙 Guides, papers, lessons, notebooks and resources for prompt engineering, context engineering, RAG, and AI Agents.

Procgen2: A community maintained fork of procgen

A simple and highly efficient RTS-game-inspired environment for reinforcement learning (formerly Gym-MicroRTS)

A tool to automate installing Atari ROMs for the Arcade Learning Environment

A collection of robotics simulation environments for reinforcement learning

An API standard for multi-agent reinforcement learning environments, with popular reference environments and related utilities

Highly Customizable Gym interface for ViZDoom.

Reinforcement Learning environments based on the 1993 game Doom

Advent of Code in pure TensorFlow

Train vision models using JAX and 🤗 transformers

Shared code for training sentence embeddings with Flax / JAX

A metrics library for the JAX ecosystem

Python sample codes and textbook for robotics algorithms.

A curated list of the latest breakthroughs in AI by release date with a clear video explanation, link to a more in-depth article, and code

A curated list of the latest breakthroughs in AI (in 2021) by release date with a clear video explanation, link to a more in-depth article, and code.

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

JAX implementation ViT-VQGAN

Turn SymPy expressions into trainable JAX expressions.

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Materials and IPython notebooks for "Python for Data Analysis" by Wes McKinney, published by O'Reilly Media

Implementation of Perceiver, General Perception with Iterative Attention

An Implementation of Transformer in Transformer in TensorFlow for image classification, attention inside local patches

scikit-learn: machine learning in Python