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
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆331Updated last week
• anas-rz / k3-addons
  Additional multi-backend functionality for Keras 3.
  ☆16Updated last year
• drisspg / driss_torch
  Cuda extensions for PyTorch
  ☆11Updated 4 months ago
• iree-org / iree-jax
  ☆52Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆292Updated 2 weeks ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 11 months ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆97Updated 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
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆414Updated 2 months ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆97Updated this week
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆12Updated last week
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆26Updated this week
• intel / intel-extension-for-openxla
  ☆50Updated 3 months ago
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆145Updated last week
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆216Updated last year
• james77777778 / keras-image-models
  A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extrac…
  ☆46Updated 8 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆201Updated last week
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆62Updated last week
• openxla / shardy
  MLIR-based partitioning system
  ☆125Updated this week
• google / aqt
  ☆328Updated this week
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆290Updated last year
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆208Updated 7 months ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆419Updated this week
• google / array_record
  ☆98Updated last week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆104Updated last year
• suo / lintrunner
  ☆28Updated last month
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆51Updated this week
• abhijangda / fastkron
  ☆19Updated 5 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆283Updated last week