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

Clear current search query, filters, and sorts
25 Open 188 Closed
25 Open 188 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

Slow training step occasionally due to slow graph flatten Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4336 opened Oct 25, 2024 by kriscao-cohere
@cgarciae
3
nnx.pmap documentation seems to be for vmap Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4330 opened Oct 24, 2024 by hrbigelow
@cgarciae
2
NNX Conv Layer Input Tuple Error Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4295 opened Oct 14, 2024 by riverliway
@8bitmp3
1
Flax dataclasses do not support new-style PRNGKeyArray in serialisation Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4130 opened Aug 19, 2024 by PhilipVinc
SPMD for initializing model using nnx.jit Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4129 opened Aug 18, 2024 by mmorinag127
2
lowering / cost analysis of @nnx.jit functions Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4094 opened Jul 19, 2024 by cgarciae
3
GroupNorm missing from NNX normalization layer Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4086 opened Jul 17, 2024 by treigerm
@cgarciae
2
feature_request: support for tabulate/summary in the NNX API Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3962 opened Jun 4, 2024 by AshishKumar4
1
with_partitioning has surprising behavior with MultiHeadAttention and DenseGeneral Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3892 opened Apr 30, 2024 by jameslyon
1
Add ability to easily change documentation version Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3858 opened Apr 15, 2024 by emergenz
@chiamp
Improve SEO for docs pages Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3854 opened Apr 12, 2024 by carlosgmartin
1
@chiamp
2
Unpickled modules with constructor arguments cannot be initialized Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3846 opened Apr 11, 2024 by eseraygun
@cgarciae
Error when calling Module.tabulate on normalization wrappers like WeightNorm and SpectralNorm Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3779 opened Mar 21, 2024 by chiamp
@cgarciae @chiamp
*Module Parameters* section of docs is outdated. Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3761 opened Mar 14, 2024 by PaulScemama
@chiamp
4
Common Initializers Does Not Work with Bias Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#2749 opened Dec 21, 2022 by daskol
1
@chiamp
6
Use enterprise test runners for our CI once they are working again Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#2731 opened Dec 16, 2022 by marcvanzee
@marcvanzee
1
better support for RNN layers required Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#2170 opened Jun 2, 2022 by sourabh2k15
@cgarciae
2
Fix missing seperator in static fold in Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required) pull ready
#2159 opened May 27, 2022 by jheek Loading…
1
@jheek
3
LazyRNG can accidentally have key sharing across layers. Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#2157 opened May 26, 2022 by levskaya
1
@jheek
2
We broke the 1:1 correspondence with attribute names and variable dict names Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#2100 opened May 4, 2022 by marcvanzee
1
@cgarciae
5
Update checkpoints.py to add a pool_size param to checkpoints.restore_checkpoint cla: yes Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required) pull ready
#1533 opened Sep 9, 2021 by mjsML Loading…
1 of 4 tasks
@levskaya
3
Implementing a custom lifted transformation: jax.experimental.ode.odeint Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#1496 opened Aug 15, 2021 by jatentaki
@levskaya
9
Improve RTD for initializers Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#1386 opened Jun 24, 2021 by billmark
@chiamp
9
Updating examples/nlp_seq Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#568 opened Oct 29, 2020 by andsteing
10 tasks
@bohnetbd
6
Improve Error Message: Assigning a list of Modules to self incorrectly Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#524 opened Oct 9, 2020 by marcvanzee
@marcvanzee @levskaya
15
ProTip! Exclude everything labeled bug with -label:bug.