Benchmarks for classification of genomic sequences

Latent Program Network (from the "Searching Latent Program Spaces" paper)

computer vision and sports

The LLM's practical guide: From the fundamentals to deploying advanced LLM and RAG apps to AWS using LLMOps best practices

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

NanoGPT (124M) quality in 2.67B tokens

NanoGPT (124M) in 3.4 minutes

A mini deep learning framework

Official inference framework for 1-bit LLMs

Implementation for MatMul-free LM.

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.

A simple and efficient Mamba implementation in pure PyTorch and MLX.

A simple, performant and scalable Jax LLM!

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

Large Context Attention

🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

Extending JAX with custom C++ and CUDA code

Mamba SSM architecture

Machine Learning Engineering Open Book

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

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

The Startup CTO's Handbook, a book covering leadership, management and technical topics for leaders of software engineering teams

Official Repository for "Eureka: Human-Level Reward Design via Coding Large Language Models" (ICLR 2024)

🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

⚡ Flashbax: Accelerated Replay Buffers in JAX

Evaluating long-term memory of reinforcement learning algorithms

Playing Pokemon Red with Reinforcement Learning

MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research