Discussions

• 1 You must be logged in to vote

  #️⃣

  v0.7.5

  levskaya asked Oct 28, 2023 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  The JIT time of my model is too long. Is there anything that can be improved?

  nasyxx asked Sep 29, 2023 in Q&A · Unanswered

  

  

  3
• 2 You must be logged in to vote

  🙏

  Inconsistent results between Flax Dense layer and Pytorch Linear

  devzhk asked May 28, 2023 in Q&A · Answered

  

  

  1
• 2 You must be logged in to vote

  🙏

  How to get per-example gradients in Flax?

  marcvanzee asked Jan 15, 2021 in Q&A · Answered

  

  

  

  

  3
• 1 You must be logged in to vote

  #️⃣

  Understanding vmap + make_rng

  JamesAllingham asked Oct 4, 2023 in General · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Convert for loop stacking, potentially with remat_scan

  mahdikooshkbaghi asked Sep 14, 2023 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  jitting configdict

  marcvanzee asked Oct 5, 2023 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  EfficientNetv2 in Flax

  faresobeid asked Sep 28, 2023 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  💡

  mish activation function.

  Nightbringers asked Oct 4, 2023 in Ideas · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Full Module RNG Scope

  joeryjoery asked Sep 28, 2023 in Q&A · Unanswered

  

  

  3
• 1 You must be logged in to vote

  🙏

  Effect of argument seq_lengths in RNN module

  Kajiih asked Sep 22, 2023 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  #️⃣

  Announcement: new signature for MultiHeadDotProductAttention's __call__ method

  chiamp asked Oct 3, 2023 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  flax speed up slow torch GPU training on large doc2vec corpus

  chrisalvino asked Oct 3, 2023 in Q&A · Closed · Unanswered

  

  

  0
• 1 You must be logged in to vote

  #️⃣

  Announcement: Flax will move from returning FrozenDicts to regular dicts

  chiamp asked Jul 11, 2023 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  💡

  Feature request: spectral normalization

  rdilip asked Feb 23, 2023 in Ideas · Closed · Answered

  

  

  

  

  3
• 1 You must be logged in to vote

  🙏

  Programmatically extend struct.dataclass class with additional attribute

  JohannesAck asked Sep 18, 2023 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  #️⃣

  v0.7.4

  cgarciae asked Sep 13, 2023 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  #️⃣

  Version 0.7.3

  IvyZX asked Sep 13, 2023 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Does flax run modules implicitly in vmap? neither jax.lax.cond or flax.nn.cond works as expected!

  dhyani15 asked Sep 12, 2023 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Feature request: Add include_init_carry to RNN

  Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.

  carlosgmartin asked Jun 17, 2023 in Q&A · Unanswered

  

  

  

  

  7
• 1 You must be logged in to vote

  🙏

  Calculating Two Gradient for two different loss in same functions

  erfanzar asked Sep 8, 2023 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  How to train sharded model using multi-host TPUs?

  Quasar-Kim asked Sep 11, 2023 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Model Parallel - Sharding Model Parameters

  peterdavidfagan asked Aug 9, 2023 in Q&A · Closed · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Simple RNN - Error: AssignSubModuleError or CallCompactUnboundModuleError

  sendeniz asked May 21, 2023 in Q&A · Unanswered

  

  

  

  

  2
• 1 You must be logged in to vote

  🙏

  Apply method of a module

  GoktugGuvercin asked Aug 28, 2023 in Q&A · Answered

  

  

  2