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

137 Open 280 Closed
137 Open 280 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
nn.compact and setup giving different results
#1767 opened Jan 7, 2022 by franchesoni
PJit example of ImageNet
#1763 opened Jan 6, 2022 by mattiasmar
Inconsistent network behaviour when using different batch sizes for model.apply on CPU
#1755 opened Jan 5, 2022 by mohamad-amin
20
Complicated aliasing issue when passing modules as arguments in transforms
#1746 opened Dec 30, 2021 by levskaya
@levskaya @jheek
1
Building JAX port of Detectron2
#1740 opened Dec 22, 2021 by INF800
9
\examples\imagenet\models_test.py fails on GPU (8 x Titan-m):
#1731 opened Dec 16, 2021 by mattiasmar
5
Checkpoints stored before 0.3.6 have incompatible GroupNorm weights
#1721 opened Dec 13, 2021 by andsteing
@jheek
Please consider lifting custom_vjp
#1713 opened Dec 11, 2021 by NeilGirdhar
1
@jheek
5
Update error message for GroupNorm default argument group_size
#1704 opened Dec 9, 2021 by sudhakarsingh27
@sudhakarsingh27
6
[Feature Request] scaling factor for initializers Priority: P2 - medium
#1688 opened Nov 29, 2021 by PhilipVinc
2
Improve Error Message: Consider extending AttributeError on Flax Modules Priority: P2 - medium
#1685 opened Nov 25, 2021 by marcvanzee Improve Error Messages
@marcvanzee
2
RNN Layers Priority: P2 - medium Status: pull requests welcome
#1677 opened Nov 23, 2021 by kamalkraj
3
Further improve Flax Modules on ReadTheDocs Priority: P2 - medium
#1676 opened Nov 17, 2021 by marcvanzee
@marcvanzee
checkpoints.restore_checkpoint` should raise an error for explicit paths that don't exist Priority: P2 - medium
#1631 opened Oct 21, 2021 by jheek
1
ogbg_molpcba example doesn't run on multi GPU machine Priority: P3 - low
#1620 opened Oct 15, 2021 by ckt624
@jheek
1
Improve documentation of key concepts behind initialization in nn.Module Priority: P1 - high
#1617 opened Oct 14, 2021 by billmark Documentation
@jheek
[Documentation]: clarify mutability of collections rather than variables Priority: P2 - medium
#1616 opened Oct 14, 2021 by levskaya Documentation
@levskaya
Need to update the LR schedule HOWTO to use Optax and TrainState Priority: P2 - medium Status: pull requests welcome
#1577 opened Sep 27, 2021 by avital
@andsteing
Add doc: "Flax Module lifecycle" Priority: P1 - high
#1576 opened Sep 27, 2021 by avital
@jheek
Feature request: Gradient checkpointing in Flax - examples or HOWTO Priority: P2 - medium
#1575 opened Sep 27, 2021 by 8bitmp3
Need to add a "lifted while_loop" Priority: P2 - medium
#1571 opened Sep 27, 2021 by avital
@jheek
3
Showcase "transfer learning" with imagenet example Priority: P2 - medium
#1567 opened Sep 24, 2021 by andsteing
@andsteing
fix checkpointing handling of giant msgpack array leaves Priority: P2 - medium
#1556 opened Sep 21, 2021 by levskaya
@levskaya
flax.training.checkpoints.save_checkpoint is blocking Priority: P3 - low
#1543 opened Sep 15, 2021 by mjsML
1
PEP484-compatible export of names in Flax Priority: P2 - medium
#1542 opened Sep 14, 2021 by manifest
1
ProTip! Adding no:label will show everything without a label.