Code

grab bag of improvements#3

Code

Open

GallagherCommaJack wants to merge 21 commits intonshepperd:masternshepperd/jaxtorch:masterfrom GallagherCommaJack:masterGallagherCommaJack/jaxtorch:masterCopy head branch name to clipboard

+1,818-240

Commits

Commits on Mar 10, 2022

• add jmp policy support to Context, Module

  

  theseriousadult

  committed
  7b2fff1

  Copy full SHA for 7b2fff1

• add support for resampling channel counts in image

  

  theseriousadult

  committed
  4cefd27

  Copy full SHA for 4cefd27

• add scale options to initializers + add glorot_uniform

  

  theseriousadult

  committed
  6340201

  Copy full SHA for 6340201

• improve initializations

  Show description for 171289b
  

  theseriousadult

  committed
  171289b

  Copy full SHA for 171289b

• import jaxatorch.init in __init__.py

  Show description for 535bc4e
  

  theseriousadult

  committed
  535bc4e

  Copy full SHA for 535bc4e

• factor out state dict conversion functions

  

  theseriousadult

  committed
  c9d7f5c

  Copy full SHA for c9d7f5c

• fix shape error in normalizer

  

  theseriousadult

  committed
  e76d74b

  Copy full SHA for e76d74b

Commits on Mar 15, 2022

• ignore pycache

  

  theseriousadult

  committed
  67773e4

  Copy full SHA for 67773e4

• add mean to normal init + add mup inits

  

  theseriousadult

  committed
  03d792e

  Copy full SHA for 03d792e

Commits on Mar 17, 2022

• fmt + force Context to be constructed with kwargs

  

  theseriousadult

  committed
  e5cef98

  Copy full SHA for e5cef98

• fix other uses of Context

  

  theseriousadult

  committed
  83c2f08

  Copy full SHA for 83c2f08

Commits on Mar 19, 2022

• add wrap_policy_cast decorator

  

  theseriousadult

  committed
  2a0fa75

  Copy full SHA for 2a0fa75

Commits on Mar 21, 2022

• add fn_wrap_policy_cast

  

  theseriousadult

  committed
  9714ffa

  Copy full SHA for 9714ffa

Commits on Mar 22, 2022

• fix unstable mean/std calculation in LN and GN

  

  theseriousadult

  committed
  531c527

  Copy full SHA for 531c527

Commits on Mar 29, 2022

• delete buggy cbor module, more initializers, more modules

  GallagherCommaJack

  committed
  3bc6785

  Copy full SHA for 3bc6785

Commits on May 1, 2022

• added poetry info

  gadgetsam

  committed
  7911deb

  Copy full SHA for 7911deb

• added more packages

  gadgetsam

  committed
  300f615

  Copy full SHA for 300f615

Commits on Sep 27, 2022

• 0.1.1

  GallagherCommaJack

  committed
  790a3c5

  Copy full SHA for 790a3c5

Commits on Dec 19, 2022

• loosen all requirements

  GallagherCommaJack

  committed
  9ac7934

  Copy full SHA for 9ac7934

Commits on May 24, 2023

• Update monkeypatches.py for compat with new jax version

  GallagherCommaJack

  authored
  71876b1

  Copy full SHA for 71876b1

Commits on Jun 7, 2023

• allow python3.11

  GallagherCommaJack

  committed
  83eeb00

  Copy full SHA for 83eeb00