Alternatives and similar repositories for keras-benchmarks

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

• drisspg / driss_torch
  Cuda extensions for PyTorch
  ☆12Updated last month
• iree-org / iree-jax
  ☆55Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆325Updated last week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆115Updated 11 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆130Updated 3 weeks ago
• chsasank / device-benchmarks
  Benchmarks of different devices I have come across
  ☆39Updated 4 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆377Updated this week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆110Updated last year
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated last month
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• anas-rz / k3-addons
  Additional multi-backend functionality for Keras 3.
  ☆16Updated last year
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆158Updated this week
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆28Updated 3 weeks ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year
• huggingface / gpu-fryer
  Where GPUs get cooked 👩‍🍳🔥
  ☆347Updated 3 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆371Updated this week
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆297Updated last year
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆105Updated this week
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆223Updated last year
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆13Updated 2 months ago
• gpu-mode / kernelbot
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆66Updated last week
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆226Updated 2 weeks ago
• meta-pytorch / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆48Updated 4 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆154Updated 2 years ago
• google / aqt
  ☆342Updated last week
• SzymonOzog / Penny
  Hand-Rolled GPU communications library
  ☆76Updated last month
• lianakoleva / no-libtorch-compile
  ☆21Updated 10 months ago
• zinccat / flaxattention
  ☆26Updated last year
• unixpickle / learn-ptx
  Learning about CUDA by writing PTX code.
  ☆151Updated last year