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: 3 months ago
Linked from 3 awesome lists
deep-learningdeep-reinforcement-learningjaxmachine-learningnumpyreinforcement-learningtransformer
Related projects:
Repository | Description | Stars |
---|---|---|
| Provides a flexible neural network library and ecosystem for JAX | 6,196 |
| A library designed to train transformer language models with reinforcement learning using various optimization techniques and fine-tuning methods. | 10,308 |
| Provides tools and libraries for training and fine-tuning large language models using transformer architectures | 6,215 |
| Implementations of various deep learning algorithms and techniques with accompanying documentation | 57,177 |
| Provides a framework for training large-scale language models on GPUs with advanced features and optimizations. | 6,997 |
| An NLP project offering various text classification models and techniques for deep learning exploration | 7,881 |
| Provides building blocks for Reinforcement Learning agents using TensorFlow | 3,136 |
| 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 |
| A 3D environment for testing and training artificial intelligence agents | 7,146 |
| A framework for distributed reinforcement learning of large language models with human feedback | 4,537 |
| Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
| A framework for training and serving large language models using JAX/Flax | 2,428 |
| Automates machine learning tasks to train accurate models in just a few lines of code | 8,167 |
| A code completion model trained on large amounts of programming language data to help developers write code more efficiently. | 6,987 |
| An open-source JavaScript library for running machine learning models in the browser without a server. | 12,363 |