DwaraknathT

thefuck

Magnificent app which corrects your previous console command.

llama.cpp

LLM inference in C/C++

system-design

Learn how to design systems at scale and prepare for system design interviews

taichi

Productive, portable, and performant GPU programming in Python.

minGPT

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models

ml-engineering

Machine Learning Engineering Open Book

picoGPT

An unnecessarily tiny implementation of GPT-2 in NumPy.

alphatensor

xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

awesome-phd-advice

Collection of advice for prospective and current PhD students

keops

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

awesome-equivariant-network

Paper list for equivariant neural network

jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

dice

A drop-in replacement of Redis with SQL-based realtime reactivity.

jaxton

100 exercises to learn JAX

tpu-starter

Everything you want to know about Google Cloud TPU

levanter

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

guacamol

Benchmarks for generative chemistry

escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

flash-cosine-sim-attention

Implementation of fused cosine similarity attention in the same style as Flash Attention

PaLM-colossalai

Scalable PaLM implementation of PyTorch

INT8-Flash-Attention-FMHA-Quantization

quick-start-connectors

This open-source repository offers reference code for integrating workplace datastores with Cohere's LLMs, enabling developers and businesses to perform seamless retrieval-augmented generation (RAG) on their own data.

Nerf.jl

mamba-jax

Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX

ML-Optimizers-JAX

Toy implementations of some popular ML optimizers using Python/JAX

GraphTransformer.jl

Graph Transformers in Flux

optimus

ML from scratch in python and CUDA for educational purposes.