☆47Feb 26, 2026Updated 8 months ago
Alternatives and similar repositories for statedict2pytree
Users that are interested in statedict2pytree are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Deprecate away!!☆16Oct 26, 2026Updated last week
- A functional training loops library for JAX☆87Feb 13, 2024Updated 2 years ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Graph neural networks in JAX.☆67Jun 18, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Run PyTorch in JAX. 🤝☆317Oct 13, 2025Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆68Jan 23, 2026Updated 9 months ago
- ☆93Aug 18, 2024Updated 2 years ago
- A simple library for scaling up JAX programs☆149Nov 4, 2025Updated last year
- A Wadler–Lindig pretty printer for Python☆49Oct 25, 2026Updated last week
- JAX Arrays for human consumption☆111Oct 23, 2025Updated last year
- ☆54May 20, 2024Updated 2 years ago
- ☆27Jul 9, 2024Updated 2 years ago
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- 3D Gaussian Splatting in JAX☆76May 30, 2024Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆60Oct 8, 2026Updated 3 weeks ago
- A simple example of VAEs with KANs☆12May 17, 2024Updated 2 years ago
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated 2 years ago
- seqax = sequence modeling + JAX☆199Jul 23, 2025Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago
- A Linux toolset for the Remarkable 2 tablet, which enables math handwriting recognition and LaTeX generation over USB via Mathpix.☆15Mar 7, 2025Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,899Jun 22, 2025Updated last year
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆12Sep 16, 2024Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆143Updated this week
- Learning astrophysics with top-k sparse feature networks (SFNets)☆11Feb 21, 2026Updated 8 months ago
- Neural Simulation-based Inference with GNN for Jeans Modeling☆10Feb 23, 2024Updated 2 years ago
- ☆14Mar 31, 2024Updated 2 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,999Oct 25, 2026Updated last week
- Minimal yet performant LLM examples in pure JAX☆286Aug 23, 2026Updated 2 months ago
- Ultra fast atomistic storage database for random reads and ML training☆13Sep 30, 2026Updated last month
- Graph Networks with Jax☆11Mar 23, 2022Updated 4 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆38Dec 3, 2023Updated 2 years ago
- Powder X-ray diffraction Rietveld refinement using FullProf.☆12Mar 20, 2022Updated 4 years ago
- Contrastive Poisson latent variable models (CPLVMs)☆11Feb 7, 2022Updated 4 years ago
- stochs: fast stochastic solvers for machine learning in C++ and Cython☆27Oct 13, 2022Updated 4 years ago
- ☆23Dec 16, 2024Updated last year
- Named Tensors for Legible Deep Learning in JAX☆232Sep 1, 2026Updated 2 months ago
- Clean Outputs from Language Models☆11May 30, 2026Updated 5 months ago