Probabilistic reasoning and statistical analysis in TensorFlow

Run PyTorch in JAX. 🤝

Awesome tmux-based terminal divider

Arrays, Tensors and dynamic Neural Networks in Mojo 🔥

Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters

OpTree: Optimized PyTree Utilities

Deep learning for dummies. All the practical details and useful utilities that go into working with real models.

Fast data store for Pandas time-series data

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

The fastest way to create an HTML app

A hyperparameter optimization framework

Add Svelte components to your Quarto documents

😎 Awesome lists about all kinds of interesting topics

DSPy: The framework for programming—not prompting—foundation models

Resources about xLSTM by Sepp Hochreiter

Visualizations of the theory behind diffusion models.

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

Composable autodiff for the interactive web.

High-speed simulator of convolutional spiking neural networks with at most one spike per neuron.

Official PyTorch Implementation of "The Hidden Attention of Mamba Models"

A tool for writing better scripts

Mamba SSM architecture

Efficiently Composable Data Augmentation on the GPU with Jax

Multiple dispatch over abstract array types in JAX.

A framework for few-shot evaluation of language models.

Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch

Quarto template for arXiv preprints

Sample project to demonstrate the simplicity of RawJS projects.