Alternatives and similar repositories for keras-benchmarks

Users that are interested in keras-benchmarks are comparing it to the libraries listed below

• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• lianakoleva / no-libtorch-compile
  ☆21Updated 8 months ago
• drisspg / driss_torch
  Cuda extensions for PyTorch
  ☆11Updated 7 months ago
• boweiliu / nccl
  Optimized primitives for collective multi-GPU communication
  ☆10Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆112Updated 10 months ago
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆12Updated 2 weeks ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆363Updated this week
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆103Updated this week
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆151Updated 3 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆108Updated last year
• iree-org / iree-jax
  ☆53Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated 2 months ago
• intel / intel-extension-for-openxla
  ☆51Updated last week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆308Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆362Updated this week
• meta-pytorch / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆47Updated 3 months ago
• suo / lintrunner
  ☆28Updated 4 months ago
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆41Updated 2 years ago
• SzymonOzog / Penny
  Hand-Rolled GPU communications library
  ☆65Updated this week
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆26Updated 2 months ago
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆61Updated last week
• huggingface / gpu-fryer
  Where GPUs get cooked 👩‍🍳🔥
  ☆317Updated 2 months ago
• IaroslavElistratov / triton-autodiff
  ☆18Updated last week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆435Updated this week
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆149Updated last week
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆68Updated 7 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆37Updated last month
• axonn-ai / axonn
  Parallel framework for training and fine-tuning deep neural networks
  ☆68Updated 2 weeks ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆110Updated last year