Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
☆37Aug 27, 2023Updated 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:
- Official PyTorch code for HILA☆28Nov 1, 2022Updated 3 years ago
- ☆12Jan 4, 2024Updated 2 years ago
- [WIP] Transformer to embed Danbooru labelsets☆13Mar 31, 2024Updated last year
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Sep 29, 2021Updated 4 years ago
- Keras 3 Implementation of Video Swin Transformers for 3D Video Modeling☆32Oct 29, 2025Updated 4 months ago
- The project scraps articles from a malayalam newspaper website to create a corpus. A set of queries is created and corresponding ground t…☆20Jan 31, 2022Updated 4 years ago
- A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extrac…☆47Jan 1, 2026Updated 2 months ago
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated 9 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- Source code of "Hold me tight! Influence of discriminative features on deep network boundaries"☆21Dec 10, 2021Updated 4 years ago
- Contains code to demonstrate distributed training in TensorFlow 2 with AI Platform and custom Docker contains.☆20Apr 28, 2021Updated 4 years ago
- ☆23Jun 18, 2024Updated last year
- ☆21Mar 3, 2025Updated last year
- 🤝 Trade any tensors over the network☆31Sep 27, 2023Updated 2 years ago
- A memory profiler for NVIDIA GPUs to explore memory inefficiencies in GPU-accelerated applications.☆27Oct 13, 2024Updated last year
- [ICLR 2025] SONICS: Synthetic Or Not - Identifying Counterfeit Songs☆44May 23, 2025Updated 9 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 9 months ago
- Code repository for the public reproduction of the language modelling experiments on "MatFormer: Nested Transformer for Elastic Inference…☆31Nov 14, 2023Updated 2 years ago
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- ResNet-RS models rewritten in Tensorflow / Keras functional API.☆27Jul 26, 2023Updated 2 years ago
- ☆34Sep 10, 2024Updated last year
- Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer☆27Dec 24, 2023Updated 2 years ago
- Unofficial implementation of "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" (https://arxiv.org/abs/2103.14030)☆67Jun 28, 2021Updated 4 years ago
- This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.☆30Aug 22, 2022Updated 3 years ago
- This is a port of Mistral-7B model in JAX☆33Jul 1, 2024Updated last year
- This repository demonstrates how to use TensorFlow based SegFormer model in 🤗 transformers package.☆30Jul 25, 2022Updated 3 years ago
- ☆33Nov 4, 2024Updated last year
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Mar 23, 2022Updated 3 years ago
- ☆34May 14, 2025Updated 9 months ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆40Dec 2, 2023Updated 2 years ago
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- Writing FLUX in Triton☆42Sep 22, 2024Updated last year
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Jul 19, 2024Updated last year
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 3 months ago
- A python algorithm to change the pitch of the voice in real time☆13Dec 13, 2020Updated 5 years ago
- Python library for Myra☆10Jan 21, 2019Updated 7 years ago
- ☆39Oct 3, 2022Updated 3 years ago