☆47Feb 26, 2026Updated last week
Alternatives and similar repositories for statedict2pytree
Users that are interested in statedict2pytree are comparing it to the libraries listed below
Sorting:
- E-book for AIS1003☆18Oct 27, 2023Updated 2 years ago
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 6 months ago
- Graph neural networks in JAX.☆68Jun 18, 2024Updated last year
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated last year
- Deprecate away!!☆16Jul 24, 2024Updated last year
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆55Jul 15, 2025Updated 7 months ago
- A Wadler–Lindig pretty printer for Python☆46Jan 16, 2026Updated last month
- Rerun c3d files, biorbd, OpenSim models from python.☆20Feb 26, 2026Updated last week
- Implementation of Bitune: Bidirectional Instruction-Tuning☆24Jun 19, 2025Updated 8 months ago
- seqax = sequence modeling + JAX☆187Jul 23, 2025Updated 7 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- ☆91Aug 18, 2024Updated last year
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- Multiple dispatch over abstract array types in JAX.☆137Dec 15, 2025Updated 2 months ago
- This SDK allows web-based apps/pages to interact with dictation devices☆17Feb 12, 2026Updated 3 weeks ago
- ☆13Oct 5, 2025Updated 5 months ago
- 💫 ManimGL with interactive previewing in VSCode☆40Apr 14, 2025Updated 10 months ago
- A simple library for scaling up JAX programs☆146Nov 4, 2025Updated 4 months ago
- Build a conda package from a setuptools project☆33Feb 5, 2026Updated last month
- You should use PySR to find scaling laws. Here's an example.☆33Sep 30, 2023Updated 2 years ago
- Code for minimum-entropy coupling.☆32Jan 6, 2026Updated 2 months ago
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated last year
- A documentation automation system for SaMD and medical device software. Documentation-as-code for ISO62304 compliant development processe…☆13Feb 8, 2026Updated last month
- Sugar makes Velodrome and Aerodrome devs life sweeter 🍭☆15Feb 4, 2026Updated last month
- A hackable Model Context Protocol (MCP) gateway☆39Updated this week
- ☆12Sep 21, 2023Updated 2 years ago
- Using JAX to generate piano music as MIDI☆39Nov 28, 2023Updated 2 years ago
- Training code for Sparse Autoencoders on Embedding models☆39Feb 27, 2025Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,870Jun 22, 2025Updated 8 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 3 weeks ago
- ☆39Apr 27, 2024Updated last year
- Flash JetPack 6 to a Jetson Orin Developer Kit☆17Feb 19, 2025Updated last year
- A ComfyUI extension that enables seamless sharing of AI-generated images and videos directly to Discord.☆12Updated this week
- CMake dependency tracking library☆13Feb 22, 2026Updated 2 weeks ago
- KHI ROS2 driver☆17Feb 18, 2026Updated 2 weeks ago
- ☆11Mar 12, 2024Updated last year
- A pre-commit hook for Pyrefly.☆22Updated this week
- Implements Global Word Vectors.☆11Feb 8, 2020Updated 6 years ago