https://github.com/google/trax

Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax

deep-learning deep-reinforcement-learning jax machine-learning numpy reinforcement-learning transformer

Last synced: 12 days ago
JSON representation

Trax — Deep Learning with Clear Code and Speed

• Host: GitHub
• URL: https://github.com/google/trax
• Owner: google
• License: apache-2.0
• Created: 2019-10-05T15:09:14.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2025-04-10T21:24:04.000Z (24 days ago)
• Last Synced: 2025-04-19T22:27:41.946Z (15 days ago)
• Topics: deep-learning, deep-reinforcement-learning, jax, machine-learning, numpy, reinforcement-learning, transformer
• Language: Python
• Homepage:
• Size: 162 MB
• Stars: 8,193
• Watchers: 137
• Forks: 824
• Open Issues: 125
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Authors: AUTHORS

Awesome Lists containing this project

• awesome-starts - google/trax - Trax — Deep Learning with Clear Code and Speed (Python)
• awesome-robotic-tooling - Trax - A library for deep learning that focuses on sequence models and reinforcement learning. (Sensor Processing / Machine Learning)
• awesome-list - Trax - Deep Learning with Clear Code and Speed (Deep Learning Framework / High-Level DL APIs)
• awesome-jax - Trax - "Batteries included" deep learning library focused on providing solutions for common workloads. <img src="https://img.shields.io/github/stars/google/trax?style=social" align="center"> (Libraries)
• awesome-python-machine-learning-resources - GitHub - 41% open · ⏱️ 08.08.2022): (Others)
• StarryDivineSky - google/trax