Pinned Loading

1. Griffin-Jax Griffin-Jax Public

   Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"

   Python 14

2. jax-ml/jax-triton jax-ml/jax-triton Public

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 425 51

3. xLSTM-Jax xLSTM-Jax Public

   Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)

   Python 17

4. Tri-RMSNorm Tri-RMSNorm Public

   Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.

   Python 11 2

5. SynthToT SynthToT Public

   SynthToT: Generate synthetic dataset for your training dataset through deliberate problem-solving et al S Yao, 2023.

   Python 9

6. mpi-ds mpi-ds Public

   MPI Operator DeepSpeed Base Configuration for CIFAR-10

   Dockerfile 4