bhosmer / mm
☆110Updated 2 months ago
Related projects: ⓘ
- ☆180Updated 2 months ago
- Automatic gradient descent☆206Updated last year
- Solve puzzles. Learn CUDA.☆53Updated 9 months ago
- ☆82Updated 6 months ago
- An interactive exploration of Transformer programming.☆243Updated 10 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (2024)☆169Updated 3 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆53Updated last month
- σ-GPT: A New Approach to Autoregressive Models☆53Updated last month
- ☆124Updated 7 months ago
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆84Updated last year
- ☆68Updated 2 months ago
- JAX implementation of the Llama 2 model☆205Updated 7 months ago
- ☆325Updated 11 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆246Updated 3 months ago
- The boundary of neural network trainability is fractal☆155Updated 7 months ago
- Gpu benchmark☆35Updated 2 weeks ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆82Updated 3 weeks ago
- Graph neural networks in JAX.☆66Updated 3 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆94Updated 2 weeks ago
- GPT-2 (124M) quality in 5B tokens☆227Updated last week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆88Updated 11 months ago
- A puzzle to learn about prompting☆106Updated last year
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆152Updated 11 months ago
- A really tiny autograd engine☆85Updated 5 months ago
- A Jax-based library for designing and training transformer models from scratch.☆272Updated 3 weeks ago
- Repository for code used in the xVal paper☆110Updated 5 months ago
- ☆141Updated last year
- Resources from the EleutherAI Math Reading Group☆50Updated 2 months ago
- ☆160Updated last year
- For optimization algorithm research and development.☆240Updated last week