PinnedLoading

1. Griffin-JaxGriffin-JaxPublic

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

   Python 14

2. jax-ml/jax-tritonjax-ml/jax-tritonPublic

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 405 50

3. xLSTM-JaxxLSTM-JaxPublic

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

   Python 17

4. Tri-RMSNormTri-RMSNormPublic

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

   Python 9 2

5. SynthToTSynthToTPublic

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

   Python 9

6. mpi-dsmpi-dsPublic

   MPI Operator DeepSpeed Base Configuration for CIFAR-10

   Dockerfile 4