Popular repositories Loading

1. tf_custom_op tf_custom_op Public

   C++

2. llama-2-jax llama-2-jax Public

   Forked from ayaka14732/llama-2-jax

   JAX implementation of the Llama 2 model

   Python

3. jax-gpu-ops jax-gpu-ops Public

4. MIOpen MIOpen Public

   Forked from ROCm/MIOpen

   AMD's Machine Intelligence Library

   Assembly

5. composable_kernel composable_kernel Public

   Forked from ROCm/composable_kernel

   Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators

   C++

6. flash-attention flash-attention Public

   Forked from ROCm/flash-attention

   Fast and memory-efficient exact attention

   Python