Code for BLT research paper

[NeurIPS'19] Deep Equilibrium Models

Official inference framework for 1-bit LLMs

This repository provides tutorials and implementations for various Generative AI Agent techniques, from basic to advanced. It serves as a comprehensive guide for building intelligent, interactive A…

Development repository for the Triton language and compiler

Get up and running with Llama 3.3, Phi 4, Gemma 2, and other large language models.

Implementation for MatMul-free LM.

Official implementation of our paper "Scalable Event-by-event Processing of Neuromorphic Sensory Signals With Deep State-Space Models"

A State-Space Model with Rational Transfer Function Representation.

A massively parallel, high-level programming language

Sparsity-aware deep learning inference runtime for CPUs

Code for our NeurIPS 2021 paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains

Open weights language model from Google DeepMind, based on Griffin.

jaxsnn is an event-based approach to machine-learning-inspired training and simulation of SNNs, including support for the BrainScaleS-2 neuromorphic backend.

Zulip-Notion linking bot that saves arxiv papers and their info when their link is posted in Zulip

IBM Analog Hardware Acceleration Kit

Some preliminary explorations of Mamba's context scaling.

A post-modern modal text editor.

A highly experimental vi-inspired editor where you edit code, not text.

Mamba SSM architecture

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

A PyTorch implementation of EventProp [https://arxiv.org/abs/2009.08378], a method to train Spiking Neural Networks

Minimalist ML framework for Rust

Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch

jax-triton contains integrations between JAX and OpenAI Triton

Event-based neural networks

Structured state space sequence models

Second Order Optimization and Curvature Estimation with K-FAC in JAX.