tensorops / TransformerX
Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow β
, Pytorch π, and Jax π)
β53Updated last year
Alternatives and similar repositories for TransformerX:
Users that are interested in TransformerX are comparing it to the libraries listed below
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"β98Updated 4 months ago
- Gradient Boosting Reinforcement Learning (GBRL)β108Updated last month
- deep learning with pytorch lightningUpdated 5 months ago
- Complete implementation of Llama2 with/without KV cache & inference πβ47Updated 11 months ago
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog poβ¦β91Updated last year
- Named Entity Recognition with an decoder-only (autoregressive) LLM using HuggingFaceβ41Updated 5 months ago
- Visualising Losses in Deep Neural Networksβ16Updated 9 months ago
- β14Updated last year
- A miniture AI training framework for PyTorchβ40Updated 2 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGOβ57Updated 6 months ago
- Gzip and nearest neighbors for text classificationβ56Updated last year
- I learn about and explain quantizationβ26Updated last year
- Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracyβ127Updated 2 years ago
- ML/DL Math and Method notesβ60Updated last year
- Training small GPT-2 style models using Kolmogorov-Arnold networks.β116Updated 11 months ago
- Interpret text data using LLMs (scikit-learn compatible).β163Updated last month
- Toy genetic algorithm in Pytorchβ39Updated this week
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.β33Updated 3 years ago
- Repository containing awesome resources regarding Hugging Face tooling.β46Updated last year
- Utilities for Training Very Large Modelsβ58Updated 7 months ago
- Notebook and Scripts that showcase running quantized diffusion models on consumer GPUsβ38Updated 5 months ago
- β54Updated 7 months ago
- making the official triton tutorials actually comprehensibleβ26Updated last month
- This repository shows various ways of deploying a vision model (TensorFlow) from π€ Transformers.β30Updated 2 years ago
- Highly commented implementations of Transformers in PyTorchβ136Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of newβ¦β120Updated 9 months ago
- RAGs: Simple implementations of Retrieval Augmented Generation (RAG) Systemsβ100Updated 3 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of modelsβ31Updated 6 months ago
- Generalist and Lightweight Model for Text Classificationβ121Updated 2 weeks ago
- A repository containing general tutorials I'd like to share with the world.β38Updated this week