A concise but complete full-attention transformer with a set of promising experimental features from various papers

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

A project exploring and reviewing the analysis and discussion of aperiodic neural activity through time.

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

State Space Models library in JAX

Implementations of various linear RNN layers using pytorch and triton

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

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

Deep memory and sequence models in JAX

Explanation to key concepts in ML

ICLR2025 Search using RAG

95.47% on CIFAR10 with PyTorch

Convolutions for Sequence Modeling

Structured state space sequence models

Building blocks for foundation models.

Long Range Arena for Benchmarking Efficient Transformers

On-device intelligence.

Find dead Python code

Diffusion models in PyTorch

Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.

A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.

Academic CVs that you can emulate

A professional list of Papers, Tutorials, and Surveys on AI for Time Series in top AI conferences and journals.

Python library for designing and training your own Diffusion Models with PyTorch.

Simulation-based inference toolkit

This repository contains research code for the paper "Generating realistic neurophysiological time series with denoising diffusion probabilistic models". @jsvetter

Refine high-quality datasets and visual AI models

Normalizing flows in PyTorch