File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -135,12 +135,12 @@ There are a few challenges we’re actively working to resolve. We encourage con
135
135
136
136
## Roadmap
137
137
138
- - [ ] Add TPU support
139
- - [ ] Optimize VRAM usage with gradient checkpointing
138
+ - [ ] Add TPU support ( https://github.com/ml-gde/jflux/issues/19 )
139
+ - [ ] Optimize VRAM usage with gradient checkpointing ( https://github.com/ml-gde/jflux/issues/20 )
140
140
- [ ] Explore further optimizations for image generation time
141
141
- [ ] Improve the handling of bfloat16 tensors with JAX
142
- - [ ] Serialise ` nnx ` weights during cli run
143
- - [ ] Port ` t5 ` and ` clip ` models to ` nnx.Module `
142
+ - [ ] Serialise ` nnx ` weights during cli run ( https://github.com/ml-gde/jflux/issues/22 )
143
+ - [ ] Port ` t5 ` and ` clip ` models to ` nnx.Module ` ( https://github.com/ml-gde/jflux/issues/21 )
144
144
145
145
## Acknowledgements
146
146
You can’t perform that action at this time.
0 commit comments