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
817 forks
Language: Python
last commit: about 1 month 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 | Provides a flexible neural network library and ecosystem for JAX | 6,196 |
 huggingface/trl | A library designed to train transformer language models with reinforcement learning using various optimization techniques and fine-tuning methods. | 10,308 |
 google-research/text-to-text-transfer-transformer | Provides tools and libraries for training and fine-tuning large language models using transformer architectures | 6,215 |
 labmlai/annotated_deep_learning_paper_implementations | Implementations of various deep learning algorithms and techniques with accompanying documentation | 57,177 |
 eleutherai/gpt-neox | Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. | 6,997 |
 brightmart/text_classification | An NLP project offering various text classification models and techniques for deep learning exploration | 7,881 |
 google-deepmind/trfl | Provides building blocks for Reinforcement Learning agents using TensorFlow | 3,136 |
| 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. | 136,357 |
| google-deepmind/lab | A 3D environment for testing and training artificial intelligence agents | 7,146 |
 carperai/trlx | A framework for distributed reinforcement learning of large language models with human feedback | 4,537 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
 young-geng/easylm | A framework for training and serving large language models using JAX/Flax | 2,428 |
 autogluon/autogluon | Automates machine learning tasks to train accurate models in just a few lines of code | 8,167 |
 deepseek-ai/deepseek-coder | A code completion model trained on large amounts of programming language data to help developers write code more efficiently. | 6,987 |
| huggingface/transformers.js | An open-source JavaScript library for running machine learning models in the browser without a server. | 12,363 |