Personal solutions to the Triton Puzzles
☆20Jul 18, 2024Updated last year
Alternatives and similar repositories for Triton-Puzzles-Solutions
Users that are interested in Triton-Puzzles-Solutions are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- We develop world models that can be adapted with natural language. Intergrating these models into artificial agents allows humans to effe…☆25Feb 10, 2024Updated 2 years ago
- Selected list of papers on World Models that I found interesting and/or useful.☆37Feb 8, 2025Updated last year
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated 10 months ago
- General Matrix Multiplication using NVIDIA Tensor Cores☆28Jan 25, 2025Updated last year
- ☆315Mar 31, 2026Updated 2 weeks ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Write a fast kernel and see how you compare against the best humans and AI on gpumode.com☆91Updated this week
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆68Updated this week
- ☆21Mar 3, 2025Updated last year
- Efficient implementation of DeepSeek Ops (Blockwise FP8 GEMM, MoE, and MLA) for AMD Instinct MI300X☆76Feb 11, 2026Updated 2 months ago
- Flash Attention in 300-500 lines of CUDA/C++☆36Aug 22, 2025Updated 7 months ago
- [ICLR 2022 Spotlight] Multi-Stage Episodic Control for Strategic Exploration in Text Games☆15Feb 8, 2026Updated 2 months ago
- Official Pytorch implementation of Chromatic Graph Transformers☆10Jun 14, 2023Updated 2 years ago
- Ἀνατομή is a PyTorch library to analyze representation of neural networks☆13Jan 31, 2024Updated 2 years ago
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆10Dec 30, 2024Updated last year
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Official code for "Federated learning for heterogeneous electronic health record systems with cost effective participant selection"☆12Feb 11, 2026Updated 2 months ago
- Don't just regulate gradients like in Muon, regulate the weights too☆32Jul 30, 2025Updated 8 months ago
- Implementation of various equivariant models in JAX☆19Apr 12, 2024Updated 2 years ago
- Implementation of Unified Embedding: Battle-Tested Feature Representations for Web-Scale ML Systems☆14Nov 11, 2023Updated 2 years ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Persistent dense gemm for Hopper in `CuTeDSL`☆15Aug 9, 2025Updated 8 months ago
- Unofficial Scalable-Softmax Is Superior for Attention☆20May 30, 2025Updated 10 months ago
- ☆12Jan 17, 2024Updated 2 years ago
- Find context neurons in Pythia models.☆13Jun 13, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code for experiments on transformers using Markovian data.☆22Nov 22, 2024Updated last year
- ☆23Apr 7, 2026Updated last week
- ☆13Jun 2, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Least Squares Regression for subspace clustering☆10May 27, 2018Updated 7 years ago
- ☆12Mar 19, 2021Updated 5 years ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- ☆57Feb 24, 2026Updated last month
- iADMM for a low-rank representation optimization problem☆13Feb 5, 2021Updated 5 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆10Aug 21, 2023Updated 2 years ago
- Conditional Linear Dynamical Systems☆16Oct 7, 2025Updated 6 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆77Updated this week
- ☆23Nov 8, 2023Updated 2 years ago
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆129Updated this week
- ☆10Apr 24, 2023Updated 2 years ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated last year