srush / Tensor-Puzzles
Solve puzzles. Improve your pytorch.
☆3,477Updated 8 months ago
Alternatives and similar repositories for Tensor-Puzzles:
Users that are interested in Tensor-Puzzles are comparing it to the libraries listed below
- Solve puzzles. Learn CUDA.☆10,714Updated 6 months ago
- What would you do with 1000 H100s...☆1,013Updated last year
- Puzzles for learning Triton☆1,508Updated 4 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,743Updated 3 months ago
- The full minitorch student suite.☆2,029Updated 7 months ago
- ☆420Updated 5 months ago
- Tile primitives for speedy kernels☆2,153Updated this week
- GPU programming related news and material links☆1,412Updated 2 months ago
- A PyTorch native library for large model training☆3,453Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,884Updated last week
- Language model alignment-focused deep learning curriculum☆1,342Updated 7 months ago
- Schedule-Free Optimization in PyTorch☆2,111Updated 2 weeks ago
- Tensors, for human consumption☆1,198Updated 3 months ago
- JAX - A curated list of resources https://github.com/google/jax☆1,725Updated 3 weeks ago
- Material for gpu-mode lectures☆4,013Updated last month
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,252Updated 3 months ago
- An autoregressive character-level language model for making more things☆2,937Updated 9 months ago
- Video+code lecture on building nanoGPT from scratch☆3,964Updated 7 months ago
- UNet diffusion model in pure CUDA☆600Updated 8 months ago
- Machine Learning Engineering Open Book☆13,141Updated last week
- The n-gram Language Model☆1,399Updated 7 months ago
- Puzzles for exploring transformers☆334Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆782Updated 2 weeks ago
- Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors a…☆1,305Updated this week
- NanoGPT (124M) in 3 minutes☆2,384Updated last week
- An unofficial styleguide and best practices summary for PyTorch☆1,964Updated 3 years ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,345Updated this week
- A simple, performant and scalable Jax LLM!☆1,653Updated this week
- PyTorch native post-training library☆4,994Updated this week
- Monte Carlo tree search in JAX☆2,440Updated 3 months ago