An interactive HTML pretty-printer for machine learning research in IPython notebooks.
☆460Aug 8, 2025Updated 6 months ago
Alternatives and similar repositories for treescope
Users that are interested in treescope are comparing it to the libraries listed below
Sorting:
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- Library for reading and processing ML training data.☆685Updated this week
- ☆270Feb 23, 2026Updated last week
- Orbax provides common checkpointing and persistence utilities for JAX users☆483Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,743Feb 16, 2026Updated 2 weeks ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 7 months ago
- ☆620Jan 23, 2026Updated last month
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- JAX - A curated list of resources https://github.com/google/jax☆2,058Jan 20, 2026Updated last month
- ☆925Jan 29, 2026Updated last month
- Evals meant to evaluate language models' ability to reason over long contexts.☆10Sep 12, 2024Updated last year
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 3 months ago
- A Python package for the statistical analysis of A/B tests☆316Feb 22, 2026Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆314Feb 25, 2026Updated last week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Jul 24, 2025Updated 7 months ago
- OpTree: Optimized PyTree Utilities☆205Feb 23, 2026Updated last week
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Optax is a gradient processing and optimization library for JAX.☆2,193Feb 24, 2026Updated last week
- ☆164Dec 13, 2023Updated 2 years ago
- A Wadler–Lindig pretty printer for Python☆45Jan 16, 2026Updated last month
- jax-triton contains integrations between JAX and OpenAI Triton☆439Feb 24, 2026Updated last week
- A simple and streamlined Python script to extract and filter links from a remote HTML resource.☆24Jan 12, 2025Updated last year
- A simple, performant and scalable Jax LLM!☆2,148Feb 24, 2026Updated last week
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Uncertainty quantification with PyTorch☆378Jan 27, 2026Updated last month
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆544Feb 16, 2026Updated 2 weeks ago
- LLM Analytics☆706Oct 19, 2024Updated last year
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆332Updated this week
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 7 months ago