octoml / Apple-M1-BERT
3X speedup over Apple’s TensorFlow plugin by using Apache TVM on M1
☆135Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for Apple-M1-BERT
- 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
- benchmarking some transformer deployments☆26Updated last year
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆193Updated 2 years ago
- ☆236Updated 3 months ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Fast sparse deep learning on CPUs☆51Updated 2 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆116Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆249Updated 2 years ago
- Torch Distributed Experimental☆116Updated 3 months ago
- Distributed preprocessing and data loading for language datasets☆39Updated 7 months ago
- Training material for IPU users: tutorials, feature examples, simple applications☆87Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated 2 weeks ago
- Home for OctoML PyTorch Profiler☆107Updated last year
- An open-source efficient deep learning framework/compiler, written in python.☆652Updated last week
- FlashAttention (Metal Port)☆386Updated last month
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆175Updated last week
- ☆47Updated 3 years ago
- Repository for the QUIK project, enabling the use of 4bit kernels for generative inference - EMNLP 2024☆173Updated 7 months ago
- Blazing fast training of 🤗 Transformers on Graphcore IPUs☆82Updated 8 months ago
- ☆64Updated 2 years ago
- TensorFlow-nGraph bridge☆137Updated 3 years ago
- ☆267Updated this week
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆129Updated 2 years ago
- Accelerate PyTorch models with ONNX Runtime☆356Updated 2 months ago
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆153Updated this week
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆234Updated last year
- Implementation of Flash Attention in Jax☆196Updated 8 months ago
- PyTorch implementation of L2L execution algorithm☆106Updated last year
- Babysit your preemptible TPUs☆84Updated last year