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 2 years ago
- ☆73Updated 2 years ago
- ☆24Updated 2 years ago
- 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
- A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extrac…☆45Updated 4 months ago
- A miniture AI training framework for PyTorch☆40Updated 2 months ago
- Shows how to do parameter ensembling using differential evolution.☆10Updated 3 years ago
- ☆58Updated last year
- Quantization of LLMs and benchmarking.☆10Updated last year
- ☆16Updated 2 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆40Updated last month
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆30Updated 5 months ago
- This is a port of Mistral-7B model in JAX☆32Updated 9 months ago
- ML/DL Math and Method notes☆60Updated last year
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- Various transformers for FSDP research☆37Updated 2 years ago
- Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights &…☆23Updated last year
- deep learning with pytorch lightningUpdated 5 months ago
- Code for NeurIPS LLM Efficiency Challenge☆57Updated last year
- Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy☆127Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- vision transformers with pytorch and pytorch lightningUpdated 5 months ago
- Template repo for Python projects, especially those focusing on machine learning and/or deep learning.☆14Updated last week
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- 🤝 Trade any tensors over the network☆30Updated last year
- ☆25Updated 2 years ago
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…☆25Updated 2 years ago