Lighteval is your all-in-one toolkit for evaluating LLMs across multiple backends

Library for industrial alignment.

Arrays, Tensors and dynamic Neural Networks in Mojo 🔥

A simple, performant and scalable Jax LLM!

[IEEE ICIP 2024] Diversifying Deep Ensembles: A Saliency Map Approach for Enhanced OOD Detection, Calibration, and Accuracy

Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton

A massively parallel, high-level programming language

Tile primitives for speedy kernels

Kolmogorov Arnold Networks

Grok open release

GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection

Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"

Official implementation of the paper "Linear Transformers with Learnable Kernel Functions are Better In-Context Models"

A library for mechanistic interpretability of GPT-style language models

Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️

Mamba SSM architecture

Stainless neural networks in JAX

Deep learning in Rust, with shape checked tensors and neural networks

⚡ A Fast, Extensible Progress Bar for Python and CLI

Transformers with Arbitrarily Large Context

Jaxpr Visualisation Tool

This is the Rust course used by the Android team at Google. It provides you the material to quickly teach Rust.

Inference Llama 2 in one file of pure 🔥

Minimalist ML framework for Rust