BERYL IDE
💎
Chat with Amanda and create 3D avatars
TJ SPENCER JACQUES PRO
AIBRUH
·
AI & ML interests
None yet
Recent Activity
updated
a Space
about 1 month ago
AIBRUH/beryl-ide
published
a Space
about 1 month ago
AIBRUH/beryl-ide
new activity
about 1 month ago
sesame/csm-1b:Request: Access Request for sesame/csm-1b
Organizations
updated
a
Space
about 1 month ago
published
a
Space
about 1 month ago
Request: Access Request for sesame/csm-1b
2
#21 opened 10 months ago
by
Sadegh69
updated
a
Space
about 2 months ago
published
a
Space
about 2 months ago
reacted to
Locutusque's
post with 🔥
2 months ago
Post
2688
🚀 AutoXLA - Accelerating Large Models on TPU
AutoXLA is an experimental library that automates the distribution, optimization, and quantization of large language models for TPUs using PyTorch/XLA. It extends the Hugging Face Transformers interface with TPU-aware features such as automatic sharding, custom attention kernels, and quantization-aware loading, making large-scale deployment and training both simpler and faster.
With quantization and Splash Attention kernels, AutoXLA achieves up to 4× speedups over standard Flash Attention implementations, significantly improving throughput for both inference and training workloads.
Whether you’re experimenting with distributed setups (FSDP, 2D, or 3D sharding) or optimizing memory via LanguageModelQuantizer, AutoXLA is built to make scaling LLMs on TPU seamless.
⚠️ Note: This is an experimental repository. Expect rough edges! Please report bugs or unexpected behavior through GitHub issues.
🔗 GitHub Repository: https://github.com/Locutusque/AutoXLA
AutoXLA is an experimental library that automates the distribution, optimization, and quantization of large language models for TPUs using PyTorch/XLA. It extends the Hugging Face Transformers interface with TPU-aware features such as automatic sharding, custom attention kernels, and quantization-aware loading, making large-scale deployment and training both simpler and faster.
With quantization and Splash Attention kernels, AutoXLA achieves up to 4× speedups over standard Flash Attention implementations, significantly improving throughput for both inference and training workloads.
Whether you’re experimenting with distributed setups (FSDP, 2D, or 3D sharding) or optimizing memory via LanguageModelQuantizer, AutoXLA is built to make scaling LLMs on TPU seamless.
⚠️ Note: This is an experimental repository. Expect rough edges! Please report bugs or unexpected behavior through GitHub issues.
🔗 GitHub Repository: https://github.com/Locutusque/AutoXLA
liked
a
model
2 months ago
