JAX refactor for networks #13

brianreicher · 2022-08-08T19:54:51Z

brianreicher · 2022-08-10T19:46:37Z

brianreicher added enhancement New feature or request maintenance Edits for general refactoring labels Aug 8, 2022

brianreicher self-assigned this Aug 8, 2022

brianreicher added a commit that referenced this issue Aug 8, 2022

Remove torch UNet #13

f19a229

brianreicher added a commit that referenced this issue Aug 8, 2022

Renaming UNet_jax to UNet #13

8b024df

brianreicher added a commit that referenced this issue Aug 9, 2022

8000 Variable renaming for UNet #13

b3b8ca4

brianreicher added a commit that referenced this issue Aug 9, 2022

Full JAX UNet implementation. Needs debugging #13

66b6b19

brianreicher added a commit that referenced this issue Aug 9, 2022

Updating UNet sampling #13

90a5df2

rhoadesScholar removed the maintenance Edits for general refactoring label Aug 10, 2022

rhoadesScholar added this to the JAX compatible milestone Aug 10, 2022

brianreicher added a commit that referenced this issue Aug 10, 2022

Debug network_test_jax.py #13

369656d

brianreicher added a commit that referenced this issue Aug 10, 2022

Fix UNet to build half model #13

95865d6

brianreicher added a commit that referenced this issue Aug 10, 2022

Debugged UNet upsampling bug and network loading #13 #14

1a8dbf9

brianreicher added a commit that referenced this issue Aug 17, 2022

Convert NLayerDiscrim to JAX #13

31df822

brianreicher added a commit that referenced this issue Aug 17, 2022

Finish initial JAX NLayerDiscrim build #13

a850a3e

brianreicher added a commit that referenced this issue Aug 17, 2022

Initial full build of JAX utils #13

354529a

brianreicher added a commit that referenced this issue Aug 25, 2022

Achieved successful JAX training #13

e4e7159

brianreicher added a commit that referenced this issue Aug 26, 2022

Relative import changes for JAX #13

8217705

brianreicher added a commit that referenced this issue Aug 26, 2022

Testing updates for NLayerDiscrim #13

8e93222

brianreicher added a commit that referenced this issue Aug 29, 2022

Denoising task in JAX #13 #14

02d0728

brianreicher added a commit that referenced this issue Aug 30, 2022

Bug fixes for channels in JAX tester #13 #14

9677a77

brianreicher added a commit that referenced this issue Aug 30, 2022

ResNet JAX updates #13

bd8db54

brianreicher added a commit that referenced this issue Aug 31, 2022

ResNet fully implemented in JAX #13

5f458c1

brianreicher added a commit that referenced this issue Sep 19, 2022

Debugging of ResNet #13

c317d83

Provide feedback