T
trax
google/trax
Trax — Deep Learning with Clear Code and Speed
★8.3kstars
Python
Apache-2.0
Updated: 2d ago
📋 Project at a Glance
Tap to expand
What's this?A deep-learning/deep-reinforcement-learning tool in the Learning & Resources category, built with Python, open-source
Who made it?Maintained by google team, 8.3K⭐ on GitHub, #297 out of 3640 in Learning & Resources
Why does it exist?With the growing demand for deep-learning and deep-reinforcement-learning in Learning & Resources, google created trax as a streamlined solution.
What can it do?Key use cases: jax, machine-learning, numpy
How to install with AI?Use an AI coding assistant (Claude Code, Cursor, Copilot) to automatically set up pip dependencies and virtual env. Follow the README — the AI handles the rest.
🔗 github.com/google/trax
🔗 github.com/google/trax
Topics
deep-learningdeep-reinforcement-learningjaxmachine-learningnumpyreinforcement-learningtransformer