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 3 years ago
- ☆24Updated 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 shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.☆30Updated 3 years ago
- Experimental CUDA kernel framework unifying typed dimensions, NVRTC JIT specialization, and ML‑guided tuning.☆43Updated this week
- Quantization of LLMs and benchmarking.☆10Updated last year
- ☆59Updated last year
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- ML/DL Math and Method notes☆64Updated last year
- ☆134Updated 2 years ago
- ☆51Updated last year
- ☆16Updated 2 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- Code for NeurIPS LLM Efficiency Challenge☆59Updated last year
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆188Updated 3 years ago
- A miniture AI training framework for PyTorch☆42Updated 9 months ago
- ☆164Updated last year
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- ☆31Updated last week
- ☆124Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆91Updated 2 years ago
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆78Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated last year
- [WIP] A 🔥 interface for running code in the cloud☆85Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆59Updated 3 years ago
- Implementation of DreamBooth in KerasCV and TensorFlow.☆88Updated 2 years ago