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
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.β12Updated last year
- This repository shows various ways of deploying a vision model (TensorFlow) from π€ Transformers.β30Updated 2 years ago
- β73Updated 2 years ago
- β24Updated 2 years ago
- Contains materials for my talk "You don't know TensorFlow".β9Updated 2 years ago
- β16Updated 2 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.β22Updated 2 years ago
- vision transformers with pytorch and pytorch lightningUpdated 5 months ago
- β58Updated last year
- Shows how to do parameter ensembling using differential evolution.β10Updated 3 years ago
- A miniture AI training framework for PyTorchβ39Updated last month
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.β38Updated 3 weeks ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.β14Updated 3 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.β33Updated 2 years ago
- π€ Trade any tensors over the networkβ30Updated last year
- Quantization of LLMs and benchmarking.β10Updated 11 months ago
- ML/DL Math and Method notesβ58Updated last year
- deep learning with pytorch lightningUpdated 4 months ago
- β14Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Moduleβ84Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog poβ¦β87Updated last year
- Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights &β¦β23Updated last year
- Various transformers for FSDP researchβ37Updated 2 years ago
- Simple and easy stable diffusion inference with LightningModule on GPU, CPU and MPS (Possibly all devices supported by Lightning).β17Updated last year
- Code for NeurIPS LLM Efficiency Challengeβ57Updated 11 months ago
- Notebooks to demonstrate TimmWrapperβ15Updated 2 months ago
- A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extracβ¦β44Updated 3 months ago
- Notebook and Scripts that showcase running quantized diffusion models on consumer GPUsβ38Updated 4 months ago
- Automatically take good care of your preemptible TPUsβ36Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of modelsβ29Updated 4 months ago