trax
Deep learning library
An end-to-end deep learning library with clear code and speed
Trax — Deep Learning with Clear Code and Speed
8k stars
140 watching
816 forks
Language: Python
last commit: 2 months ago
Linked from 3 awesome lists
deep-learningdeep-reinforcement-learningjaxmachine-learningnumpyreinforcement-learningtransformer
christoschristofidis/awesome-deep-learning
ly0n/awesome-robotic-tooling
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google/flax | A high-performance neural network library and ecosystem designed for flexibility | 6,132 |
 huggingface/trl | A library designed to train transformer language models with reinforcement learning using various optimization techniques and fine-tuning methods. | 10,053 |
 google-research/text-to-text-transfer-transformer | Provides tools and libraries for training and fine-tuning large language models using transformer architectures | 6,170 |
 labmlai/annotated_deep_learning_paper_implementations | Implementations of various deep learning algorithms and techniques with accompanying documentation | 56,215 |
 eleutherai/gpt-neox | Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. | 6,941 |
 brightmart/text_classification | An NLP project offering various text classification models and techniques for deep learning exploration | 7,861 |
 google-deepmind/trfl | Provides building blocks for Reinforcement Learning agents using TensorFlow | 3,134 |
| huggingface/transformers | 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. | 135,022 |
| google-deepmind/lab | A 3D environment for testing and training artificial intelligence agents | 7,133 |
 carperai/trlx | A framework for distributed reinforcement learning of large language models with human feedback | 4,502 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
 young-geng/easylm | A framework for training and serving large language models using JAX/Flax | 2,409 |
 autogluon/autogluon | An automated machine learning framework that simplifies the process of building accurate predictive models from raw data, using minimal code and configuration. | 8,039 |
 deepseek-ai/deepseek-coder | A code completion model trained on large amounts of programming language data to help developers write code more efficiently. | 6,837 |
| huggingface/transformers.js | An API for using pre-trained machine learning models in web browsers without the need for a server | 12,085 |