Skip to content

Commit 8dd99cc

Browse files
docs: add issues ref
1 parent e307cf8 commit 8dd99cc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,12 +135,12 @@ There are a few challenges we’re actively working to resolve. We encourage con
135135

136136
## Roadmap
137137

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)
140140
- [ ] Explore further optimizations for image generation time
141141
- [ ] 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)
144144

145145
## Acknowledgements
146146

0 commit comments

Comments
 (0)