IntelLabs / SLIDE_opt_iaLinks
ā74Updated 2 years ago
Alternatives and similar repositories for SLIDE_opt_ia
Users that are interested in SLIDE_opt_ia are comparing it to the libraries listed below
Sorting:
- Nod.ai š¦ version of š» . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository ā¦ā106Updated last week
- PyTorch interface for the IPUā181Updated 2 years ago
- benchmarking some transformer deploymentsā26Updated this week
- š Pytorch code for the Nero optimiser.ā20Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.ā29Updated 4 years ago
- Python Research Frameworkā106Updated 3 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodesā242Updated 2 years ago
- A GPT, made only of MLPs, in Jaxā58Updated 4 years ago
- SLIDE (Sub-LInear Deep learning Engine) written in Goā45Updated 5 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.ā149Updated 2 years ago
- Customized matrix multiplication kernelsā57Updated 3 years ago
- ā40Updated 2 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)ā189Updated 3 years ago
- ā71Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorchā184Updated 2 years ago
- a lightweight transformer library for PyTorchā72Updated 4 years ago
- Differentiable Algorithms and Algorithmic Supervision.ā116Updated 2 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight ā¦ā236Updated 2 years ago
- PyTorch implementation of L2L execution algorithmā109Updated 2 years ago
- An implementation of Additive Attentionā150Updated 3 years ago
- Productionize machine learning predictions, with ONNX or withoutā66Updated last year
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementationā29Updated 9 months ago
- Torch Distributed Experimentalā117Updated last year
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)ā118Updated 3 years ago
- Pytorch Lightning Distributed Accelerators using Rayā215Updated 2 years ago
- Lightweight machine learning library based on OpenCL 1.2ā75Updated 5 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters iā¦ā182Updated 3 months ago
- Texture mapping with variational auto-encodersā40Updated 4 years ago
- Hacks for PyTorchā19Updated 2 years ago
- Stride visualizationsā38Updated 7 years ago