790gradml / gradMLLinks
6.790 | Machine Learning | Draft Site/Notes
☆14Updated last month
Alternatives and similar repositories for gradML
Users that are interested in gradML are comparing it to the libraries listed below
Sorting:
- ☆16Updated last year
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆71Updated 8 months ago
- all the materials for cs140e winter 2026☆33Updated this week
- Typst extension, adding support for generating figures using inline Python code☆113Updated last month
- General Matrix Multiplication using NVIDIA Tensor Cores☆28Updated last year
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆157Updated 2 months ago
- Train across all your devices, ezpz 🍋☆26Updated last week
- Can RL solve simple problems?☆54Updated 2 years ago
- Smaller and faster nanochat in MLX☆36Updated 2 months ago
- Minimal, beautiful (+ highly-customizable) styles for Matplotlib.☆19Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Updated 2 years ago
- ☆45Updated 3 months ago
- JAX implementation of the Mistral 7b v0.2 model☆35Updated last year
- Files needed to replicate my research blog☆95Updated last year
- SIMD quantization kernels☆94Updated 4 months ago
- Minimal Implimentation of VCRec (2024) for collapse provention.☆17Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆69Updated 3 months ago
- An interactive guide to the Fourier-Transform.☆26Updated last year
- ☆138Updated last year
- Running Jax in PyTorch Lightning☆119Updated last year
- NYU Artificial Intelligence Spring 2024☆61Updated last year
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆36Updated 4 months ago
- Neural Networks for JAX☆84Updated last year
- TUI for the Slurm Workload Manager☆419Updated last week
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆75Updated 7 months ago
- A functional training loops library for JAX☆88Updated last year
- Because we don't want a jupyter notebook mess...☆61Updated 7 months ago
- An implementation of delta-iris in tinygrad☆72Updated last year
- Minimal yet performant LLM examples in pure JAX☆236Updated 2 weeks ago
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆276Updated 3 months ago