← all repositories

kyegomez/zeta

A modular PyTorch framework providing reusable high-performance components for building AI models.

595 stars Python ML FrameworksLanguage Models
zeta
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

Zeta is a library of modular building blocks for AI model development in PyTorch. It provides carefully optimized components including attention mechanisms (multi-query attention, flash attention, sigmoid attention), mixture of experts routing, feedforward networks, normalization layers, and other transformer architecture components. Developers can assemble state-of-the-art models by combining these pre-built, tested modules rather than implementing them from scratch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.