Just Layers over @jax-ml

A Survey of Reinforcement Learning for Large Reasoning Models

A minimal implementation of DeepMind's Genie world model

Open-source framework for the research and development of foundation models.

Optax is a gradient processing and optimization library for JAX.

A JAX research toolkit for building, editing, and visualizing neural networks.

The lean application framework for Python. Build sophisticated user interfaces with a simple Python API. Run your apps in the terminal and a web browser.

Multiple dispatch in Python

MLIR-based partitioning system

A JAX-native LLM Post-Training Library

Library for reading and writing large multi-dimensional arrays.

Implement minimal boilerplate CLIs derived from type hints and parse from command line, config files and environment variables

Rewrap extension for VSCode and Visual Studio

Einops, now in Julia

Julia Implementation of Low Memory Disk ANN (LM-DiskANN)

All-in-Storage Solution based on DiskANN for DRAM-free Approximate Nearest Neighbor Search

Faster way to switch between clusters and namespaces in kubectl

Library for reading and processing ML training data.

A user-space file system for interacting with Google Cloud Storage

Official Implementation of "Affordable AI Assistants with Knowledge Graph of Thoughts"

Orbax provides common checkpointing and persistence utilities for JAX users

Optimize Julia Functions With MLIR and XLA for High-Performance Execution on CPU, GPU, TPU and more.

A simple, performant and scalable Jax LLM!

An Extensible Deep Learning Library

It slices, it dices, it splices!

Metal programming in Julia

Reading and writing of video files in Julia via ffmpeg

Julia library for deep reinforcement learning

AI Observability & Evaluation

Distributed Compiler based on Triton for Parallel Systems