google/traxArchived

Features

• Attention Optimization - Efficient transformer implementation using hash attention.
• Deep Learning Frameworks - Library for deep learning research and training.
• Neural Network Frameworks - Efficient transformer architecture implementation.
• Neural Network Libraries - Batteries-included library for common deep learning workloads.
• Perception and Machine Learning - Deep learning library for sequence and reinforcement models.