Pinned Loading
-
Griffin-Jax
Griffin-Jax PublicJax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
Python 12
-
jax-ml/jax-triton
jax-ml/jax-triton Publicjax-triton contains integrations between JAX and OpenAI Triton
-
Tri-RMSNorm
Tri-RMSNorm PublicEfficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.
Something went wrong, please refresh the page to try again.
If the problem persists, check the GitHub status page or contact support.
If the problem persists, check the GitHub status page or contact support.