t5x

Sequence model trainer

A modular framework for training and deploying sequence models at scale

GitHub

3k stars
36 watching
308 forks
Language: Python
last commit: 21 days 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,170
tensorflow/tpu Provides reference models and tools for training machine learning models on Cloud TPUs. 5,213
eleutherai/gpt-neox Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. 6,941
higgsfield-ai/higgsfield A framework for efficient and fault-tolerant distributed training of large neural networks on multiple GPUs. 3,293
google-research/big_vision Supports large-scale vision model training on GPU machines or Google Cloud TPUs using scalable input pipelines. 2,334
google/brax A physics simulation framework designed for research and development in robotics, reinforcement learning, and other fields. 2,337
huggingface/text-generation-inference A toolkit for deploying and serving Large Language Models. 9,106
triton-inference-server/server Provides an optimized cloud and edge inferencing solution for AI models 8,342
google/trax An end-to-end deep learning library with clear code and speed 8,096
epistasislab/tpot Automated machine learning tool that optimizes machine learning pipelines using genetic programming 9,736
tensorflow/serving A high-performance serving system for machine learning models in production environments. 6,185
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,356
carperai/trlx A framework for distributed reinforcement learning of large language models with human feedback 4,502
great-expectations/great_expectations Provides tools and techniques to ensure data quality by defining expected outcomes for data processing pipelines. 9,989