IntelLabs / SLIDE_opt_ia
☆74Updated 10 months ago
Related projects: ⓘ
- benchmarking some transformer deployments☆26Updated last year
- Customized matrix multiplication kernels☆53Updated 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 …☆107Updated this week
- hyperparameter management☆43Updated last year
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- ☆18Updated last year
- PyTorch implementation of L2L execution algorithm☆107Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated last year
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆236Updated last year
- A tracing JIT compiler for PyTorch☆12Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated last year
- Implementation of a Tensorflow XLA rematerialization pass☆15Updated 4 years ago
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- Python Research Framework☆107Updated last year
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- ☆40Updated this week
- ☆12Updated 3 years ago
- High performance pytorch modules☆18Updated last year
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆147Updated last year
- Torch Distributed Experimental☆115Updated last month
- a lightweight transformer library for PyTorch☆71Updated 2 years ago
- A "gym" style toolkit for building lightweight NAS systems.☆13Updated 2 years ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated last year
- Hacks for PyTorch☆17Updated last year
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆48Updated 9 months ago
- Block-sparse primitives for PyTorch☆147Updated 3 years ago
- SLIDE (Sub-LInear Deep learning Engine) written in Go☆41Updated 4 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year