IntelLabs / SLIDE_opt_iaLinks
☆74Updated last year
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:
- ☆40Updated 2 years ago
- 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 8 months ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241Updated 2 years ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆43Updated 2 years ago
- benchmarking some transformer deployments☆26Updated 2 years ago
- Python Research Framework☆106Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆117Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- PyTorch interface for the IPU☆181Updated last year
- Lightweight machine learning library based on OpenCL 1.2☆75Updated 4 years ago
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆189Updated 3 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- SLIDE (Sub-LInear Deep learning Engine) written in Go☆45Updated 5 years ago
- A stateful pytree library for training neural networks.☆22Updated 3 weeks ago
- tensor4 - pytorch to C++ convertor using lightweight templated tensor library☆28Updated 5 years ago
- Productionize machine learning predictions, with ONNX or without☆66Updated last year
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- Deep learning for the Web☆38Updated 4 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☆108Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆56Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 3 years ago