Alternatives and similar repositories for keras-xla-benchmarks

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

• jiaweizzhao / ZerO-initialization
  ☆75Updated 2 years ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆42Updated 6 months ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 3 years ago
• Ujjawal-K-Panchal / coshnet
  ☆51Updated last year
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• sayakpaul / cait-tf
  Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.
  ☆12Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 3 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 3 years ago
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆57Updated 3 years ago
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆51Updated last year
• apple / ml-mofi
  ☆59Updated last year
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆25Updated 6 months ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• rasbt / cvpr2023
  ☆134Updated last year
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆35Updated 2 weeks ago
• AnswerDotAI / minai
  A miniture AI training framework for PyTorch
  ☆42Updated 6 months ago
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆166Updated 6 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• lucidrains / AMIE-pytorch
  Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind
  ☆67Updated 11 months ago
• landskape-ai / Progressive-Pruning
  Official pytorch code for "APP: Anytime Progressive Pruning" (DyNN @ ICML, 2022; CLL @ ACML, 2022, SNN @ ICML, 2022 and SlowDNN 2023)
  ☆16Updated 2 years ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆124Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• Lightning-AI / utilities
  Common Python utilities and GitHub Actions in Lightning Ecosystem
  ☆57Updated last week
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• epfml / DenseFormer
  ☆82Updated last year
• eth-easl / fmengine
  Utilities for Training Very Large Models
  ☆58Updated 11 months ago