t5x
Sequence model trainer
A modular framework for training and deploying sequence models at scale
3k stars
36 watching
309 forks
Language: Python
last commit: 3 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
| google-research/text-to-text-transfer-transformer | Provides tools and libraries for training and fine-tuning large language models using transformer architectures | 6,215 |
 tensorflow/tpu | Provides reference models and tools for training machine learning models on Cloud TPUs. | 5,221 |
 eleutherai/gpt-neox | Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. | 6,997 |
 higgsfield-ai/higgsfield | A framework for efficient and fault-tolerant distributed training of large neural networks on multiple GPUs. | 3,299 |
| google-research/big_vision | Supports large-scale vision model training on GPU machines or Google Cloud TPUs using scalable input pipelines. | 2,439 |
 google/brax | A physics simulation framework designed for research and development in robotics, reinforcement learning, and other fields. | 2,397 |
 huggingface/text-generation-inference | A toolkit for deploying and serving Large Language Models (LLMs) for high-performance text generation | 9,456 |
 triton-inference-server/server | An open-source software that enables deployment of AI models from multiple deep learning and machine learning frameworks on various devices | 8,460 |
| google/trax | An end-to-end deep learning library with clear code and speed | 8,114 |
 epistasislab/tpot | Automated machine learning tool that optimizes machine learning pipelines using genetic programming | 9,776 |
| tensorflow/serving | A high-performance serving system for machine learning models in production environments. | 6,195 |
 tensorpack/tensorpack | A high-performance neural network training interface for TensorFlow that focuses on speed and flexibility. | 6,303 |
 google-deepmind/mctx | An open-source library providing efficient implementations of search algorithms for reinforcement learning | 2,377 |
 carperai/trlx | A framework for distributed reinforcement learning of large language models with human feedback | 4,537 |
 great-expectations/great_expectations | Provides tools and techniques to ensure data quality by defining expected outcomes for data processing pipelines. | 10,054 |