apple / parameterized-transformsLinks
torchvision-based transforms that provide access to parameterization
☆14Updated 5 months ago
Alternatives and similar repositories for parameterized-transforms
Users that are interested in parameterized-transforms are comparing it to the libraries listed below
Sorting:
- Personal solutions to the Triton Puzzles☆19Updated 11 months ago
- ☆12Updated last week
- Make triton easier☆47Updated last year
- 🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.☆11Updated last month
- LLM attention pattern visualizer☆10Updated last year
- Library for text-to-text regression, applicable to any input string representation and allows pretraining and fine-tuning over multiple r…☆86Updated this week
- ☆13Updated 2 weeks ago
- Learn CUDA with PyTorch☆29Updated this week
- Because it's there.☆16Updated 9 months ago
- ☆48Updated 8 months ago
- LLM training in simple, raw C/CUDA☆15Updated 7 months ago
- Simple repository for training small reasoning models☆33Updated 5 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 9 months ago
- ☆53Updated last year
- Using FlexAttention to compute attention with different masking patterns☆44Updated 9 months ago
- Training hybrid models for dummies.☆25Updated 6 months ago
- ☆45Updated last year
- ML/DL Math and Method notes☆61Updated last year
- Minimum Description Length probing for neural network representations☆18Updated 5 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆46Updated this week
- ☆73Updated 2 months ago
- LM engine is a library for pretraining/finetuning LLMs☆59Updated this week
- ☆51Updated last year
- Experiment of using Tangent to autodiff triton☆79Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 4 months ago
- ☆20Updated last year
- ☆18Updated last year
- Samples of good AI generated CUDA kernels☆84Updated last month
- Latent Large Language Models☆18Updated 10 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆77Updated last year