Alternatives and similar repositories for jax-scalify

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

• anh-tong / nanoGPT-equinox

  View on GitHub
  nanoGPT using Equinox
  ☆15Mar 3, 2023Updated 2 years ago
• packquickly / schedule_free_optx

  View on GitHub
  Schedule free optimiser implemented in JAX using Optimistix
  ☆15May 29, 2024Updated last year
• apple / ml-dataset-decomposition

  View on GitHub
  Official repo of dataset-decomposition paper [NeurIPS 2024]
  ☆21Jan 8, 2025Updated last year
• layer6ai-labs / calo-forest

  View on GitHub
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆19Nov 3, 2024Updated last year
• GallagherCommaJack / modulax

  View on GitHub
  ☆18Aug 24, 2024Updated last year
• NVlabs / PerAda

  View on GitHub
  Repo for the paper: PerAda: Parameter-Efficient Federated Learning Personalization with Generalization Guarantees (CVPR 2024)
  ☆23Aug 14, 2024Updated last year
• edwardmilsom / function-space-learning-rates-paper

  View on GitHub
  Code for the paper "Function-Space Learning Rates"
  ☆25Jun 3, 2025Updated 8 months ago
• davisyoshida / haiku-mup

  View on GitHub
  A port of muP to JAX/Haiku
  ☆25Oct 23, 2022Updated 3 years ago
• cloneofsimo / efae

  View on GitHub
  ☆23Jun 18, 2024Updated last year
• apple / ml-entropy-reconstruction

  View on GitHub
  ☆30Jul 5, 2023Updated 2 years ago
• CerebrasResearch / Sparse-IFT

  View on GitHub
  Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency
  ☆25Jul 31, 2024Updated last year
• iwiwi / epochraft

  View on GitHub
  Checkpointable dataset utilities for foundation model training
  ☆32Jan 29, 2024Updated 2 years ago
• Clybius / Personalized-Optimizers

  View on GitHub
  A collection of niche / personally useful PyTorch optimizers with modified code.
  ☆27Oct 25, 2025Updated 3 months ago
• lucaslingle / mu_transformer

  View on GitHub
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Jun 5, 2025Updated 8 months ago
• jdeschena / sdtt

  View on GitHub
  [ICLR 2025] SDTT: a simple and effective distillation method for discrete diffusion models
  ☆47Sep 10, 2025Updated 5 months ago
• google-research / precondition

  View on GitHub
  ☆31Jan 23, 2026Updated 3 weeks ago
• vijaymalav564 / vortex-clang

  View on GitHub
  // clone this repo with --depth=1 to save disk size // toolchain compatible with Ubuntu 20.04+ //
  ☆15Apr 28, 2022Updated 3 years ago
• LuLuLuyi / LongHeads

  View on GitHub
  [EMNLP'24] LongHeads: Multi-Head Attention is Secretly a Long Context Processor
  ☆31Apr 8, 2024Updated last year
• capjamesg / sam-clip

  View on GitHub
  Use Grounding DINO, Segment Anything, and CLIP to label objects in images.
  ☆35Dec 27, 2023Updated 2 years ago
• bronemos / view-fusion

  View on GitHub
  Official implementation of ViewFusion: Learning Composable Diffusion Models for Novel View Synthesis
  ☆36May 30, 2025Updated 8 months ago
• google-research-datasets / QAmeleon

  View on GitHub
  QAmeleon introduces synthetic multilingual QA data using PaLM, a 540B large language model. This dataset was generated by prompt tuning P…
  ☆35Aug 15, 2023Updated 2 years ago
• OpenLMLab / LongWanjuan

  View on GitHub
  Towards Systematic Measurement for Long Text Quality
  ☆37Sep 5, 2024Updated last year
• epfml / schedules-and-scaling

  View on GitHub
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆89Oct 30, 2024Updated last year
• philippe-eecs / vitok

  View on GitHub
  ☆34May 14, 2025Updated 9 months ago
• IPBench / IPBench

  View on GitHub
  Repository of IPBench
  ☆19Jan 4, 2026Updated last month
• pprablanc / ppsrt

  View on GitHub
  A python algorithm to change the pitch of the voice in real time
  ☆13Dec 13, 2020Updated 5 years ago
• LastBotInc / llama2j

  View on GitHub
  Pure Java Llama2 inference with optional multi-GPU CUDA implementation
  ☆13Sep 2, 2023Updated 2 years ago
• akhilkedia / TranformersGetStable

  View on GitHub
  [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"
  ☆10Jul 19, 2024Updated last year
• WeihuangLin / INF-LLaVA

  View on GitHub
  INF-LLaVA: Dual-perspective Perception for High-Resolution Multimodal Large Language Model
  ☆42Aug 4, 2024Updated last year
• OldPranoy / awesome-flet

  View on GitHub
  An awesome list that curates the best Flet tools, tutorials, blogs and more.
  ☆10Jan 8, 2023Updated 3 years ago
• rapidsai / legate-boost

  View on GitHub
  GBM implementation on Legate
  ☆14Jan 28, 2026Updated 2 weeks ago
• hhu-stups / prob-rodinplugin

  View on GitHub
  (READ ONLY MIRROR) The ProB Model Checker and Animator Plugin for Rodin
  ☆19Jan 24, 2026Updated 2 weeks ago
• hcmlab / audioFX

  View on GitHub
  ☆16Jul 23, 2023Updated 2 years ago
• ShuhongChen / vtubers-dataset

  View on GitHub
  CVPR 2023: PAniC-3D, Vtubers dataset downloader
  ☆13Apr 22, 2023Updated 2 years ago
• hengck23 / solution-vesuvius-challenge-ink-detection

  View on GitHub
  ☆11Jul 17, 2023Updated 2 years ago
• zhwm / SparsePro

  View on GitHub
  A fine-mapping method integrating GWAS summary statistics and functional annotation data
  ☆11Dec 28, 2023Updated 2 years ago
• Baran-phys / Tropical-Attention

  View on GitHub
  [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"
  ☆23Oct 23, 2025Updated 3 months ago
• lucidrains / grokfast-pytorch

  View on GitHub
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Dec 22, 2024Updated last year
• shuyanzhou / wikihow_hierarchy

  View on GitHub
  The source code and the data for ACL 2022 paper "Show Me More Details: Discovering Hierarchies of Procedures from Semi-structured Web Dat…
  ☆14Apr 21, 2023Updated 2 years ago