Alternatives and similar repositories for keras-benchmarks

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

• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆301Updated this week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• drisspg / driss_torch
  Cuda extensions for PyTorch
  ☆11Updated 5 months ago
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆12Updated last month
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆26Updated this week
• anas-rz / k3-addons
  Additional multi-backend functionality for Keras 3.
  ☆16Updated last year
• lianakoleva / no-libtorch-compile
  ☆21Updated 7 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆122Updated 3 weeks ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆348Updated 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 …
  ☆102Updated this week
• iree-org / iree-jax
  ☆52Updated last year
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆105Updated last year
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆148Updated last month
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆107Updated 8 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆426Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• chsasank / device-benchmarks
  Benchmarks of different devices I have come across
  ☆33Updated last month
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆144Updated 6 months ago
• google-deepmind / tf2jax
  ☆115Updated last month
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆108Updated last year
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆87Updated this week
• google / drjax
  ☆15Updated last week
• axonn-ai / axonn
  Parallel framework for training and fine-tuning deep neural networks
  ☆65Updated 6 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆207Updated 8 months ago
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆41Updated 2 years ago
• google / praxis
  ☆189Updated 2 weeks ago
• mila-iqia / milabench
  Repository of machine learning benchmarks
  ☆42Updated this week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆357Updated this week
• argonne-lcf / Megatron-DeepSpeed
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆16Updated last week