A machine learning compiler for GPUs, CPUs, and ML accelerators

A JAX-native LLM Post-Training Library

A simple, performant and scalable Jax LLM!

The best ChatGPT that $100 can buy.

SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms

Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.

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

A terminal spreadsheet multitool for discovering and arranging data

Community plugins list, theme list, and releases of Obsidian.

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

Website to track people, organizations, and products (tools, websites, etc.) in AI safety

General-purpose programming language and toolchain for maintaining robust, optimal, and reusable software.

RL Environments in JAX 🌍

Multi-Agent Experiments in JAX

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent

Multi-Agent Reinforcement Learning with JAX

Custom plugins for https://github.com/saulpw/visidata/

Enable Next-Gen Large Language Model Applications. Join our Discord: https://discord.gg/pAbnFJrkgZ

Python client library for Mistral AI platform