sayakpaul / keras-xla-benchmarksLinks
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
☆37Updated 2 years ago
Alternatives and similar repositories for keras-xla-benchmarks
Users that are interested in keras-xla-benchmarks are comparing it to the libraries listed below
Sorting:
- ☆75Updated 2 years ago
- ☆24Updated 3 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆42Updated 6 months ago
- This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.☆30Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated 2 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 3 years ago
- Code for NeurIPS LLM Efficiency Challenge☆59Updated last year
- ☆16Updated 2 years ago
- ☆134Updated last year
- ☆51Updated last year
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆57Updated 3 years ago
- Quantization of LLMs and benchmarking.☆10Updated last year
- ☆59Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- https://blog.tensorflow.org/2021/12/continuous-adaptation-for-machine.html☆29Updated 3 years ago
- Contains code to demonstrate distributed training in TensorFlow 2 with AI Platform and custom Docker contains.☆20Updated 4 years ago
- ML/DL Math and Method notes☆63Updated last year
- Latest Weight Averaging (NeurIPS HITY 2022)☆31Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind☆68Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated 2 years ago
- High-performance, asynchronous Python HTTP client library designed for faster file transfers using concurrency, semaphores, and fault-tol…☆56Updated 4 months ago
- Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and te…☆42Updated last year
- ☆15Updated 3 years ago
- ☆31Updated 3 months ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago