Alternatives and similar repositories for jax-layers:

Users that are interested in jax-layers are comparing it to the libraries listed below

• nateraw / huggingface-datasets-converter
  Scripts to convert datasets from various sources to Hugging Face Datasets.
  ☆57Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• deep-diver / complete-mlops-system-workflow
  ☆18Updated 2 years ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆108Updated 3 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆60Updated last year
• jax-ml / jax-tpu-embedding
  ☆17Updated this week
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 10 months ago
• huggingface / hffs
  **ARCHIVED** Filesystem interface to 🤗 Hub
  ☆58Updated 2 years ago
• rwightman / genalog
  Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and te…
  ☆42Updated last year
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• deep-diver / TFX-WandB
  ☆14Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 9 months ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• sgugger / torchdynamo-tests
  ☆19Updated 2 years ago
• sayakpaul / count-tokens-hf-datasets
  This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…
  ☆25Updated 2 years ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 2 years ago
• huggingface / fuego
  [WIP] A 🔥 interface for running code in the cloud
  ☆85Updated 2 years ago
• IBM / dolomite-engine
  Dolomite Engine is a library for pretraining/finetuning LLMs
  ☆49Updated this week
• mlops-discord / talks
  Slides and recordings of talks hosted by our community
  ☆20Updated 10 months ago
• jaymody / picoBERT
  Like picoGPT but for BERT.
  ☆49Updated 2 years ago
• AnswerDotAI / minai
  A miniture AI training framework for PyTorch
  ☆40Updated 2 months ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆51Updated 9 months ago
• huggingface / disaggregators
  🤗 Disaggregators: Curated data labelers for in-depth analysis.
  ☆65Updated 2 years ago
• AI4Bharat / FBI
  FBI: Finding Blindspots in LLM Evaluations with Interpretable Checklists
  ☆28Updated last month
• soumik12345 / tf2_gans
  Implementations of GANs in Tensorflow 2.x
  ☆15Updated 3 years ago
• pacman100 / peft-codegen-25
  ☆24Updated last year
• NielsRogge / awesome-huggingface
  Repository containing awesome resources regarding Hugging Face tooling.
  ☆46Updated last year
• basetenlabs / Workshop-TRT-LLM
  ☆18Updated 10 months ago
• huggingface / competitions
  ☆123Updated 5 months ago