trax
by google
Trax — Deep Learning with Clear Code and Speed
AI summary
Deep learning library
An end-to-end deep learning library with clear code and speed
- stars
- 8.1K
- forks
- 817
- watching
- 140
- awesome lists
- 3
Similar projects
Found by comparing what the projects do, not just their names.
Neural Network Library
Provides a flexible neural network library and ecosystem for JAX
huggingface/trl10.3K
Transformer model trainer
A library designed to train transformer language models with reinforcement learning using various optimization techniques and fine-tuning methods.
Transformer library
Provides tools and libraries for training and fine-tuning large language models using transformer architectures
Deep learning implementations
Implementations of various deep learning algorithms and techniques with accompanying documentation
Language model trainer
Provides a framework for training large-scale language models on GPUs with advanced features and optimizations.
NLP toolkit
An NLP project offering various text classification models and techniques for deep learning exploration
RL framework
Provides building blocks for Reinforcement Learning agents using TensorFlow
Model repository
A collection of pre-trained machine learning models for various natural language and computer vision tasks, enabling developers to fine-tune and deploy these models on their own projects.
AI testbed
A 3D environment for testing and training artificial intelligence agents
carperai/trlx4.5K
RL Framework
A framework for distributed reinforcement learning of large language models with human feedback
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
LLM framework
A framework for training and serving large language models using JAX/Flax
AutoML tool
Automates machine learning tasks to train accurate models in just a few lines of code
Code completion tool
A code completion model trained on large amounts of programming language data to help developers write code more efficiently.
Model runner
An open-source JavaScript library for running machine learning models in the browser without a server.
Featured in 3 awesome lists
Each link jumps to the spot where the list mentions trax.
