Alternatives and similar repositories for statedict2pytree

Users that are interested in statedict2pytree are comparing it to the libraries listed below

• jmtomczak / vae_kan_example

  View on GitHub
  A simple example of VAEs with KANs
  ☆11May 17, 2024Updated last year
• mancusolab / traceax

  View on GitHub
  Stochastic trace estimation using JAX
  ☆17Aug 20, 2025Updated 5 months ago
• markaren / E-book_cpp

  View on GitHub
  E-book for AIS1003
  ☆18Oct 27, 2023Updated 2 years ago
• itsdaniele / jeometric

  View on GitHub
  Graph neural networks in JAX.
  ☆68Jun 18, 2024Updated last year
• evanatyourservice / llm-jax

  View on GitHub
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆18Jul 24, 2025Updated 6 months ago
• dodgebc / jaxkd

  View on GitHub
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆53Jul 15, 2025Updated 7 months ago
• MatX-inc / seqax

  View on GitHub
  seqax = sequence modeling + JAX
  ☆170Jul 23, 2025Updated 6 months ago
• patrick-kidger / wadler_lindig

  View on GitHub
  A Wadler–Lindig pretty printer for Python
  ☆44Jan 16, 2026Updated last month
• pyomeca / pyorerun

  View on GitHub
  Rerun c3d files, biorbd, OpenSim models from python.
  ☆20Jan 29, 2026Updated 2 weeks ago
• dkopi / Bitune

  View on GitHub
  Implementation of Bitune: Bidirectional Instruction-Tuning
  ☆23Jun 19, 2025Updated 7 months ago
• thecharlieblake / lovely-llama

  View on GitHub
  An implementation of the Llama architecture, to instruct and delight
  ☆21May 31, 2025Updated 8 months ago
• RobertCsordas / moeut

  View on GitHub
  ☆91Aug 18, 2024Updated last year
• ltgoslo / elc-bert

  View on GitHub
  ☆21Apr 14, 2025Updated 10 months ago
• cgarciae / simple_flow_matching

  View on GitHub
  ☆24Dec 16, 2024Updated last year
• jax-ml / jax-llm-examples

  View on GitHub
  Minimal yet performant LLM examples in pure JAX
  ☆242Jan 14, 2026Updated last month
• DannyTran123 / egopet

  View on GitHub
  Official implementation of the paper "EgoPet: Egomotion and Interaction Data from an Animal's Perspective".
  ☆29Dec 15, 2025Updated 2 months ago
• patrick-kidger / quax

  View on GitHub
  Multiple dispatch over abstract array types in JAX.
  ☆136Dec 15, 2025Updated 2 months ago
• GoogleChromeLabs / dictation_support

  View on GitHub
  This SDK allows web-based apps/pages to interact with dictation devices
  ☆16Feb 6, 2026Updated last week
• microsoft / CtP

  View on GitHub
  ☆46Apr 30, 2021Updated 4 years ago
• ASEM000 / kernex

  View on GitHub
  Stencil computations in JAX
  ☆70Nov 15, 2025Updated 3 months ago
• ssokota / mec

  View on GitHub
  Code for minimum-entropy coupling.
  ☆32Jan 6, 2026Updated last month
• MilesCranmer / pysr_scaling_laws

  View on GitHub
  You should use PySR to find scaling laws. Here's an example.
  ☆33Sep 30, 2023Updated 2 years ago
• yixiaoer / mistral-v0.2-jax

  View on GitHub
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Jul 3, 2024Updated last year
• rajiviyer / genai_evaluation

  View on GitHub
  ☆12Sep 21, 2023Updated 2 years ago
• velodrome-finance / sugar-sdk

  View on GitHub
  Sugar makes Velodrome and Aerodrome devs life sweeter 🍭
  ☆15Feb 4, 2026Updated last week
• yoavgur / PISCES

  View on GitHub
  🪝PISCES - Precise In-Parameter Suppression for Concept EraSure in Large Language Models
  ☆12May 30, 2025Updated 8 months ago
• openCUC / openCUC

  View on GitHub
  ☆11Jul 1, 2020Updated 5 years ago
• thinkexist1989 / rocos-app

  View on GitHub
  The application of Robotics for ROCOS
  ☆11Jan 8, 2026Updated last month
• roboticslab-uc3m / yarp-devices

  View on GitHub
  A place for YARP devices
  ☆10Oct 28, 2025Updated 3 months ago
• MeindertN / RoboClerk

  View on GitHub
  A documentation automation system for SaMD and medical device software. Documentation-as-code for ISO62304 compliant development processe…
  ☆13Feb 8, 2026Updated last week
• enjalot / latent-sae

  View on GitHub
  Training code for Sparse Autoencoders on Embedding models
  ☆39Feb 27, 2025Updated 11 months ago
• facebook / pyrefly-pre-commit

  View on GitHub
  A pre-commit hook for Pyrefly.
  ☆23Updated this week
• agijsberts / ssgpr

  View on GitHub
  Batch and incremental Sparse Spectrum Gaussian Process for Regression
  ☆10Mar 16, 2021Updated 4 years ago
• Miksus / thesis-computational-artificial-market

  View on GitHub
  Artificial stock market (ASM) with Julia language.
  ☆10Aug 10, 2021Updated 4 years ago
• AEmotionStudio / ComfyUI-DiscordSend

  View on GitHub
  A ComfyUI extension that enables seamless sharing of AI-generated images and videos directly to Discord.
  ☆11Updated this week
• VDA5050 / vd-m-a-5050

  View on GitHub
  JSON Schemas for the use with the VD(M)A 5050 protocol
  ☆10Mar 21, 2022Updated 3 years ago
• ucf-photovoltaics / UCF-EL-Defect

  View on GitHub
  ☆17Apr 25, 2023Updated 2 years ago
• domluna / Glove.jl

  View on GitHub
  Implements Global Word Vectors.
  ☆11Feb 8, 2020Updated 6 years ago
• RoboticImaging / NOCaL

  View on GitHub
  NOCaL: Calibration-Free Semi-Supervised Learning of Odometry and Camera Intrinsics
  ☆11May 12, 2024Updated last year