t5x

Sequence model trainer

A modular framework for training and deploying sequence models at scale

GitHub

3k stars
36 watching
309 forks
Language: Python
last commit: almost 2 years ago

Related projects:

RepositoryDescriptionStars
google-research/text-to-text-transfer-transformerProvides tools and libraries for training and fine-tuning large language models using transformer architectures6,215
tensorflow/tpuProvides reference models and tools for training machine learning models on Cloud TPUs.5,221
eleutherai/gpt-neoxProvides a framework for training large-scale language models on GPUs with advanced features and optimizations.6,997
higgsfield-ai/higgsfieldA framework for efficient and fault-tolerant distributed training of large neural networks on multiple GPUs.3,299
google-research/big_visionSupports large-scale vision model training on GPU machines or Google Cloud TPUs using scalable input pipelines.2,439
google/braxA physics simulation framework designed for research and development in robotics, reinforcement learning, and other fields.2,397
huggingface/text-generation-inferenceA toolkit for deploying and serving Large Language Models (LLMs) for high-performance text generation9,456
triton-inference-server/serverAn open-source software that enables deployment of AI models from multiple deep learning and machine learning frameworks on various devices8,460
google/traxAn end-to-end deep learning library with clear code and speed8,114
epistasislab/tpotAutomated machine learning tool that optimizes machine learning pipelines using genetic programming9,776
tensorflow/servingA high-performance serving system for machine learning models in production environments.6,195
tensorpack/tensorpackA high-performance neural network training interface for TensorFlow that focuses on speed and flexibility.6,303
google-deepmind/mctxAn open-source library providing efficient implementations of search algorithms for reinforcement learning2,377
carperai/trlxA framework for distributed reinforcement learning of large language models with human feedback4,537
great-expectations/great_expectationsProvides tools and techniques to ensure data quality by defining expected outcomes for data processing pipelines.10,054