A simple, performant and scalable Jax LLM!

JAX-Toolbox

A post-modern modal text editor.

Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.

Powerful text refactoring with the Helix Editor

Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"

Tokamax: A GPU and TPU kernel library.

A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its variants as the primary backbone with support for ImageNet train…

Inference code for LLaMA models in JAX

Optax is a gradient processing and optimization library for JAX.

Library for reading and processing ML training data.

Reference PyTorch implementation and models for DINOv3

A comprehensive list of papers for the definition of World Models and using World Models for General Video Generation, Embodied AI, and Autonomous Driving, including papers, codes, and related webs…

PIX is an image processing library in JAX, for JAX.

verl: Volcano Engine Reinforcement Learning for LLMs

Python 3.8+ toolbox for submitting jobs to Slurm

PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838

NVIDIA Isaac Sim™ is an open-source application on NVIDIA Omniverse for developing, simulating, and testing AI-driven robots in realistic virtual environments.

PyTorch code and models for VJEPA2 self-supervised learning from video.

An extremely fast Python type checker and language server, written in Rust.

Terminal image viewer with native support for iTerm and Kitty

Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch

A non-saturating, open-ended environment for evaluating LLMs in Factorio

Production-tested AI infrastructure tools for efficient AGI development and community-driven innovation

A PyTorch native platform for training generative AI models

Efficient Triton Kernels for LLM Training