tensorops / TransformerX
Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)
☆51Updated 11 months ago
Alternatives and similar repositories for TransformerX:
Users that are interested in TransformerX are comparing it to the libraries listed below
- Gradient Boosting Reinforcement Learning (GBRL)☆95Updated last month
- Train fastai models faster (and other useful tools)☆64Updated 7 months ago
- ☆130Updated last year
- RAGs: Simple implementations of Retrieval Augmented Generation (RAG) Systems☆88Updated 9 months ago
- ML/DL Math and Method notes☆57Updated last year
- ☆42Updated 5 months ago
- ☆58Updated 10 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆95Updated 3 weeks ago
- Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.☆36Updated last year
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆178Updated 3 weeks ago
- Running Jax in PyTorch Lightning☆83Updated last month
- A miniture AI training framework for PyTorch☆37Updated 3 weeks ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆112Updated 3 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆54Updated 3 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆47Updated 7 months ago
- Gzip and nearest neighbors for text classification☆56Updated last year
- Repository containing awesome resources regarding Hugging Face tooling.☆46Updated last year
- A multi-backend (TensorFlow, PyTorch, JAX, and NumPy) implementation of the Segment Anything model in Keras 3.0☆31Updated 9 months ago
- ☆43Updated 2 months ago
- Implementation of DreamBooth in KerasCV and TensorFlow.☆87Updated last year
- Use Grounding DINO, Segment Anything, and CLIP to label objects in images.☆23Updated last year
- Utilities for Training Very Large Models☆57Updated 3 months ago
- ☆83Updated 3 months ago
- Code for NeurIPS LLM Efficiency Challenge☆54Updated 9 months ago
- 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
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆77Updated 6 months ago
- Unofficial JAX implementations of deep learning research papers☆152Updated 2 years ago
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 2 years ago