Skip to content

Issues: google/flax

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

202 Open 757 Closed
202 Open 757 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

nnx.Embed(num_embedding=1) incompatible with "jax_disable_jit" and "debug_nans"
#4557 opened Feb 20, 2025 by anthony-Neo
Improve support for optax.GradientTransformationExtraArgs in NNX
#4545 opened Feb 13, 2025 by NKlug
3
AttributeError: module 'jax._src.config' has no attribute 'enable_memories'
#4535 opened Feb 9, 2025 by jeromemassot
2
Composable modules
#4534 opened Feb 8, 2025 by danjenson
1
Restoring a checkpoint with Orbax CheckpointManager fails
#4521 opened Feb 1, 2025 by bednarikjan
1
jax.eval_shape modifies nnx.Module state when using nnx.vmap and nnx.scan
#4520 opened Feb 1, 2025 by ytsmiling
2
[linen] Undocumented RNN errors
#4513 opened Jan 29, 2025 by MRiabov
1
Why does LSTMCell keep rngs in its state?
#4509 opened Jan 28, 2025 by JoaoAparicio
1
How to unroll a model end to end?
#4507 opened Jan 27, 2025 by JoaoAparicio
3
nnx.make_causal_mask() usage
#4505 opened Jan 25, 2025 by windmaple
3
How to use (jax) pytrees inside of nnx modules?
#4497 opened Jan 22, 2025 by PhilipVinc
4
Pool functions reduce over batch dimension and not last dimension
#4494 opened Jan 21, 2025 by simonschoelly
1
Update dependencies for flaxlib
#4491 opened Jan 19, 2025 by Bot-wxt1221
Failed build: CI - with JAX nightly build failed
#4488 opened Jan 18, 2025 by github-actions bot
5
Improve error message when user mistakenly holds a jax Array in an nnx.Module
#4480 opened Jan 13, 2025 by RaghuSpaceRajan
1
2
Questions on Using nnx.value_and_grad for Loss Calculation and Model Decoupling in Flax NNX
#4476 opened Jan 10, 2025 by Tomato-toast
4
Support for exporting models in the ONNX format
#4430 opened Dec 12, 2024 by Artoriuz
2
NNX WeightNorm Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment. Status: pull requests welcome We agree with the direction proposed, feel free to give it a shot and file a pull request
#4426 opened Dec 10, 2024 by rbflx
2
Issues checkpointing optimizer state using Optax, nnx.Optimizer, and Orbax
#4423 opened Dec 9, 2024 by SandSnip3r
7
[FR] Save/load of model parameters
#4422 opened Dec 8, 2024 by PhilipVinc
2
Clarification on sharding strategy to combine multiple training steps together via nnx.scan
#4417 opened Dec 4, 2024 by Teculos
2
MNIST NNX Tutorial broken links
#4394 opened Nov 22, 2024 by stergiosba
1
5
Issue with Optimizer Update in A2C Network with Optax Body:
#4391 opened Nov 20, 2024 by Tomato-toast
6
ResNet-18 Initialization
#4388 opened Nov 19, 2024 by ZedongPeng
Flax NNX and Orbax Checkpointing require hacking to work together
#4383 opened Nov 15, 2024 by hdrwilkinson
3
ProTip! Exclude everything labeled bug with -label:bug.