wiegerw / nervaLinks
C++ and Python libraries for neural networks.
☆18Updated last month
Alternatives and similar repositories for nerva
Users that are interested in nerva are comparing it to the libraries listed below
Sorting:
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Updated last year
- Evolution Pretraining Fully in Int Formats☆134Updated 3 weeks ago
- Sparsity support for PyTorch☆38Updated 9 months ago
- ☆55Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 3 years ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Updated last year
- seqax = sequence modeling + JAX☆169Updated 5 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- A Data-Centric Compiler for Machine Learning☆85Updated 3 weeks ago
- Scalable training and inference for Probabilistic Circuits☆89Updated 2 weeks ago
- A MAD laboratory to improve AI architecture designs 🧪☆136Updated last year
- train with kittens!☆63Updated last year
- Write a fast kernel and run it on Discord. See how you compare against the best!☆66Updated 3 weeks ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Updated last year
- Einsum-like high-level array sharding API for JAX☆34Updated last year
- ☆62Updated last year
- LLM training in simple, raw C/CUDA☆109Updated last year
- Collection of kernels written in Triton language☆174Updated 9 months ago
- TORCH_LOGS parser for PT2☆70Updated last week
- Named Tensors for Legible Deep Learning in JAX☆215Updated 2 months ago
- ☆40Updated 2 years ago
- ☆28Updated 11 months ago
- Hierarchical Associative Memory User Experience☆106Updated 2 months ago
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆30Updated last year
- ☆233Updated 11 months ago
- ☆16Updated 2 months ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- ☆14Updated 2 years ago
- Learning Universal Predictors☆81Updated last year