sayakpaul / keras-xla-benchmarks
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
☆37Updated last year
Alternatives and similar repositories for keras-xla-benchmarks:
Users that are interested in keras-xla-benchmarks are comparing it to the libraries listed below
- Contains materials for my talk "You don't know TensorFlow".☆9Updated last year
- A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extrac…☆42Updated 2 months ago
- ☆72Updated 2 years ago
- Efficient CUDA kernels for training convolutional neural networks with PyTorch.☆39Updated 2 weeks ago
- ☆24Updated 2 years ago
- This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.☆29Updated 2 years ago
- Shows how to do parameter ensembling using differential evolution.☆10Updated 3 years ago
- Official pytorch code for "APP: Anytime Progressive Pruning" (DyNN @ ICML, 2022; CLL @ ACML, 2022, SNN @ ICML, 2022 and SlowDNN 2023)☆17Updated 2 years ago
- ☆16Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆29Updated 3 months ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated last year
- Notebooks to demonstrate TimmWrapper☆15Updated last month
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 2 years ago
- Code for NeurIPS LLM Efficiency Challenge☆55Updated 10 months ago
- ☆58Updated 11 months ago
- ML/DL Math and Method notes☆58Updated last year
- This is a port of Mistral-7B model in JAX☆31Updated 7 months ago
- 🤝 Trade any tensors over the network☆30Updated last year
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Various transformers for FSDP research☆35Updated 2 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆54Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆36Updated last year
- Notebook and Scripts that showcase running quantized diffusion models on consumer GPUs☆38Updated 3 months ago
- Learn CUDA with PyTorch☆16Updated 2 weeks ago
- Experiment of using Tangent to autodiff triton☆75Updated last year
- Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights &…☆23Updated last year
- ☆14Updated last year