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

Roadmap #1

Closed
4 tasks done
ariG23498 opened this issue Sep 2, 2024 · 0 comments · Fixed by #3, #5, #6, #7 or #9
Closed
4 tasks done

Roadmap #1

ariG23498 opened this issue Sep 2, 2024 · 0 comments · Fixed by #3, #5, #6, #7 or #9
Assignees
@ariG23498 @SauravMaheshkar
Labels
feature 🚀 New feature or request
Milestone
v0.0.01

Comments

@ariG23498
Copy link
Collaborator

ariG23498 commented Sep 2, 2024 •
edited
Loading

Roadmap:

Some pointers to note:

  1. Use the flax.nnx API
  2. Be very careful with the dtype mismatches -- probably will cause tests to fail
  3. Strengthen reproducibility with JAX's PRNG apis
@SauravMaheshkar SauravMaheshkar added the feature 🚀 New feature or request label Sep 2, 2024
@SauravMaheshkar SauravMaheshkar added this to the v0.0.01 milestone Sep 2, 2024
@SauravMaheshkar SauravMaheshkar mentioned this issue Sep 2, 2024
Closed
@SauravMaheshkar SauravMaheshkar mentioned this issue Oct 1, 2024
Merged
ariG23498 pushed a commit that referenced this issue Oct 2, 2024
@SauravMaheshkar
feat: use nnx.Sequential over ModuleList (#5)
5da49d7 
Part of #1
This was linked to pull requests Oct 9, 2024
feat: initial nnx port #3
Merged
feat: use nnx.Sequential over ModuleList #5
Merged
[Test] Adds tests for autoencoders #6
Merged
Adding layers and math #7
Merged
Fix the model #9
Merged
AE Porting Code and Cleanup #10
Merged
Adding FLUX porting code #11
Merged
feat: chex-ify testsuite #12
Merged
Denoiser works #13
Merged
T5 and Clip are Flax models #15
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@ariG23498 ariG23498

@SauravMaheshkar SauravMaheshkar

Labels
feature 🚀 New feature or request
Projects
None yet
Milestone
v0.0.01
2 participants
@ariG23498 @SauravMaheshkar