grab bag of improvements#3

Open

GallagherCommaJack wants to merge 21 commits intonshepperd:masterfrom

GallagherCommaJack:master

Commits on Mar 10, 2022

add jmp policy support to Context, Module
theseriousadult
committed
add support for resampling channel counts in image
theseriousadult
committed
add scale options to initializers + add glorot_uniform
theseriousadult
committed
improve initializations
theseriousadult
committed
import jaxatorch.init in __init__.py
theseriousadult
committed
factor out state dict conversion functions
theseriousadult
committed
fix shape error in normalizer
theseriousadult
committed

Commits on Mar 15, 2022

ignore pycache
theseriousadult
committed
add mean to normal init + add mup inits
theseriousadult
committed

Commits on Mar 17, 2022

fmt + force Context to be constructed with kwargs
theseriousadult
committed
fix other uses of Context
theseriousadult
committed

Commits on Mar 19, 2022

add wrap_policy_cast decorator
theseriousadult
committed

Commits on Mar 21, 2022

add `fn_wrap_policy_cast`
theseriousadult
committed

Commits on Mar 22, 2022

fix unstable mean/std calculation in LN and GN
theseriousadult
committed

Commits on Mar 29, 2022

delete buggy cbor module, more initializers, more modules
GallagherCommaJack
committed

Commits on May 1, 2022

Commits on Sep 27, 2022

0.1.1
GallagherCommaJack
committed

Commits on Dec 19, 2022

loosen all requirements
GallagherCommaJack
committed

Commits on May 24, 2023

Update monkeypatches.py for compat with new jax version
GallagherCommaJack
authored

Commits on Jun 7, 2023

allow python3.11
GallagherCommaJack
committed