Stars
Reliable, minimal and scalable library for pretraining foundation and world models
Reliable, minimal and scalable library for evaluating and conducting world model research
Minimal, lightweight JAX implementations of popular models.
Pure MLX implementations of UMAP, t-SNE, PaCMAP, TriMap, DREAMS, CNE, MMAE, and NNDescent for Apple Silicon. Metal GPU for computation and video rendering.
Minimal Claude Code alternative. Single Python file, zero dependencies, ~250 lines.
Code, data and weights for the paper **What drives success in physical planning with Joint-Embedding Predictive World Models?**
A simple, performant and scalable Jax LLM!
Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"
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…
Sea-Snell / JAX_llama
Forked from meta-llama/llamaInference 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.