paquiteau / hydra-callbacks
A collection of usefull hydra callbacks
☆12Updated 5 months ago
Alternatives and similar repositories for hydra-callbacks:
Users that are interested in hydra-callbacks are comparing it to the libraries listed below
- A Wadler--Lindig pretty printer for Python☆32Updated 2 weeks ago
- Tools for JAX☆45Updated last week
- Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing☆18Updated 3 months ago
- A framework for conducting machine learning experiments in python☆38Updated 2 months ago
- Exponential families for JAX☆63Updated last week
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago
- JAX Arrays for human consumption☆90Updated last year
- ☆59Updated 2 years ago
- Running Jax in PyTorch Lightning☆86Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆104Updated 7 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- A generic interface for linear algebra backends☆71Updated 7 months ago
- ☆100Updated 7 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆103Updated 3 months ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Image augmentation library for Jax☆37Updated 10 months ago
- 🧱 Modula software package☆139Updated this week
- Lightweight ML Experiment Logging 📖☆78Updated 5 months ago
- Agustinus' very opiniated publication-ready plotting library☆61Updated 2 weeks ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated last month
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- A library for calibrating classifiers and computing calibration metrics☆13Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated 3 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆14Updated 7 months ago
- Typst extension, adding support for generating figures using inline Python code☆77Updated 9 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆183Updated 2 months ago
- Stencil computations in JAX☆70Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆28Updated last week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆100Updated last year
- If it quacks like a tensor...☆56Updated 3 months ago