Alternatives and similar repositories for keras-benchmarks

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

• Lightning-AI / LitModels
  Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-gr…
  ☆30Updated last week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆62Updated this week
• iree-org / iree-jax
  ☆52Updated 10 months ago
• drisspg / driss_torch
  Cuda extensions for PyTorch
  ☆11Updated 2 months ago
• mila-iqia / milabench
  Repository of machine learning benchmarks
  ☆36Updated 3 weeks ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆94Updated this week
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆19Updated 11 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 3 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆96Updated 5 months ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆143Updated 4 months ago
• suo / lintrunner
  ☆28Updated last week
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆12Updated last week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆114Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆46Updated 3 weeks ago
• google / praxis
  ☆186Updated 3 weeks ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆181Updated last week
• google-deepmind / tf2jax
  ☆114Updated this week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆134Updated last year
• zinccat / flaxattention
  ☆18Updated 8 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆213Updated last year
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated 11 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆314Updated this week
• Lightning-AI / utilities
  Common Python utilities and GitHub Actions in Lightning Ecosystem
  ☆56Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆97Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆125Updated last week