Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

JAX refactor for networks #13

Open
3 of 4 tasks
brianreicher opened this issue Aug 8, 2022 · 1 comment
Open
3 of 4 tasks

JAX refactor for networks #13

brianreicher opened this issue Aug 8, 2022 · 1 comment
Assignees
@brianreicher
Labels
enhancement New feature or request
Milestone
JAX compatible

Comments

@brianreicher
Copy link
Collaborator

brianreicher commented Aug 8, 2022 •
edited
Loading

Refactor UNet, Residual UNet, MIRNet2D, and Utils from PyTorch to JAX:
@brianreicher brianreicher added enhancement New feature or request maintenance Edits for general refactoring labels Aug 8, 2022
@brianreicher brianreicher self-assigned this Aug 8, 2022
brianreicher added a commit that referenced this issue Aug 8, 2022
@brianreicher
Remove torch UNet #13
f19a229
brianreicher added a commit that referenced this issue Aug 8, 2022
@brianreicher
Renaming UNet_jax to UNet #13
8b024df
brianreicher added a commit that referenced this issue Aug 9, 2022
@brianreicher
Variable renaming for UNet #13
b3b8ca4
brianreicher added a commit that referenced this issue Aug 9, 2022
@brianreicher
Full JAX UNet implementation. Needs debugging #13
66b6b19
brianreicher added a commit that referenced this issue Aug 9, 2022
@brianreicher
Updating UNet sampling #13
90a5df2
@rhoadesScholar rhoadesScholar removed the maintenance Edits for general refactoring label Aug 10, 2022
@rhoadesScholar rhoadesScholar added this to the JAX compatible milestone Aug 10, 2022
brianreicher added a commit that referenced this issue Aug 10, 2022
@brianreicher
Debug network_test_jax.py #13
369656d
brianreicher added a commit that referenced this issue Aug 10, 2022
@brianreicher
Fix UNet to build half model #13
95865d6
@brianreicher
Copy link
Collaborator Author

brianreicher commented Aug 10, 2022 •
edited
Loading

  • Fix incorrect integer call in UNet upsampling layer

brianreicher added a commit that referenced this issue Aug 10, 2022
@brianreicher
Debugged UNet upsampling bug and network loading #13 #14
1a8dbf9
brianreicher added a commit that referenced this issue Aug 17, 2022
@brianreicher
Convert NLayerDiscrim to JAX #13
31df822
brianreicher added a commit that referenced this issue Aug 17, 2022
@brianreicher
Finish initial JAX NLayerDiscrim build #13
a850a3e
brianreicher added a commit that referenced this issue Aug 17, 2022
@brianreicher
Initial full build of JAX utils #13
354529a
brianreicher added a commit that referenced this issue Aug 25, 2022
@brianreicher
Achieved successful JAX training #13
e4e7159
brianreicher added a commit that referenced this issue Aug 26, 2022
@brianreicher
Relative import changes for JAX #13
8217705
brianreicher added a commit that referenced this issue Aug 26, 2022
@brianreicher
Testing updates for NLayerDiscrim #13
8e93222
brianreicher added a commit that referenced this issue Aug 29, 2022
@brianreicher
Denoising task in JAX #13 #14
02d0728
brianreicher added a commit that referenced this issue Aug 30, 2022
@brianreicher
Bug fixes for channels in JAX tester #13 #14
9677a77
brianreicher added a commit that referenced this issue Aug 30, 2022
@brianreicher
ResNet JAX updates #13
bd8db54
brianreicher added a commit that referenced this issue Aug 31, 2022
@brianreicher
ResNet fully implemented in JAX #13
5f458c1
brianreicher added a commit that referenced this issue Sep 19, 2022
@brianreicher
Debugging of ResNet #13
c317d83
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@brianreicher brianreicher

Labels
enhancement New feature or request
Projects
None yet
Milestone
JAX compatible
Development

No branches or pull requests
2 participants
@rhoadesScholar @brianreicher