☆81Mar 17, 2026Updated this week
Alternatives and similar repositories for init2winit
Users that are interested in init2winit are comparing it to the libraries listed below
Sorting:
- ☆31Sep 17, 2024Updated last year
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 2 weeks ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Oct 21, 2022Updated 3 years ago
- Graph Learning with JAX☆14Jul 11, 2022Updated 3 years ago
- ☆27Mar 12, 2026Updated last week
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆416Updated this week
- ☆26Updated this week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- ☆18Mar 20, 2022Updated 3 years ago
- Composable metric reporters in Python.☆14Jun 6, 2024Updated last year
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Nov 4, 2024Updated last year
- ☆194Mar 10, 2026Updated last week
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- ☆45Nov 17, 2025Updated 4 months ago
- ☆136Mar 6, 2026Updated last week
- Arete is an experimental programming language.☆12Oct 6, 2023Updated 2 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,030Dec 17, 2025Updated 3 months ago
- Every Call is Precious: Global Optimization of Black-Box Functions with Unknown Lipschitz Constants☆16Nov 30, 2025Updated 3 months ago
- ☆29Feb 21, 2026Updated 3 weeks ago
- Keras-like APIs for JAX framework☆50Mar 25, 2023Updated 2 years ago
- ☆13Feb 12, 2026Updated last month
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆551Updated this week
- Echo Noise Channel for Exact Mutual Information Calculation☆17Jul 17, 2020Updated 5 years ago
- ☆17Dec 6, 2019Updated 6 years ago
- Python package for inference with Gaussian processes☆11Mar 25, 2015Updated 10 years ago
- ☆15Apr 20, 2024Updated last year
- ☆376Mar 9, 2026Updated last week
- ☆17Jul 23, 2023Updated 2 years ago
- ☆11Dec 31, 2021Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- Python utilities for working with FHIR, including libraries to build simple, flat FHIR views in BigQuery.☆99Feb 24, 2026Updated 3 weeks ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- ☆39Updated this week
- Regularization, Neural Network Training Dynamics☆14Jan 13, 2020Updated 6 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Nov 4, 2024Updated last year
- General-purpose library and robotics framwork used by Coding with Chrome.☆24Apr 9, 2024Updated last year
- Make the contents of public meetings searchable and discoverable.☆33Jun 17, 2024Updated last year
- A python implementation of tile coding using numpy.☆11May 13, 2017Updated 8 years ago