Discussions

• 2 You must be logged in to vote

  #️⃣

  Performance considerations for NNX

  cgarciae asked Sep 24, 2024 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Are there any hidden side effects of having a list of jnp.ndarray as members to a struct.PyTreeNode

  sudo-michael asked Sep 25, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Implementation of DLRM: Embedding Operations

  Sir-NoChill asked Sep 24, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Correct initializer of the model's params with input of different sizes

  TheGlitch2701 asked Sep 20, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Difference between dtype and precision arguments?

  amishra791 asked Sep 19, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  #️⃣

  Named Tensors in NNX

  wlib asked Sep 19, 2024 in General · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  nnx.Dropout in combination with pmap returns an error

  maxxxzdn asked Sep 11, 2024 in Q&A · Closed · Answered

  

  

  2
• 7 You must be logged in to vote

  🙏

  Reduce the logging/verbosity of orbax async checkpointing

  zaccharieramzi asked Jun 5, 2023 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  L1/L2 regularization of network weights with NNX

  pushkar5586 asked Aug 30, 2024 in Q&A · Answered

  

  

  

  

  3
• 1 You must be logged in to vote

  🙏

  How to scan over list input?

  py4 asked Aug 23, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  How to efficiently use nnx.jit for methods?

  cisprague asked Sep 4, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  #️⃣

  Is nnx.vmap a replacement for jax.vmap? Getting different behavior

  jlperla asked Aug 26, 2024 in General · Answered

  

  

  3
• 1 You must be logged in to vote

  🙏

  Regarding train_step in NNX

  qnixsynapse asked Jul 10, 2024 in Q&A · Answered

  

  

  

  

  3
• 1 You must be logged in to vote

  🙏

  nnx.Dropout in combination with jax.lax.scan returns an error

  maxencefaldor asked Aug 28, 2024 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  How to return the indices of max values with max pooling?

  LauriKurki asked Jul 26, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Porting PyTorch weight_norm, trouble with Flax kernel_init for WeightNorm

  DBraun asked Aug 19, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Get index in flax linen scan

  jakubMitura14 asked Jun 6, 2023 in Q&A · Answered

  

  

  

  

  5
• 1 You must be logged in to vote

  🙏

  Flax Conv issue

  NITHISHM2410 asked Jul 31, 2024 in Q&A · Unanswered

  

  

  0
• 3 You must be logged in to vote

  🙏

  using flax sow inside a function with scanning

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

  TalKachman asked Feb 28, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Proper way to serialize and load model structure

  JeyRunner asked Jul 18, 2024 in Q&A · Unanswered

  

  

  6
• 1 You must be logged in to vote

  🙏

  Large Model output (bsz x seq_len x vocab_size) repeated 3 times in memory with jax.value_and_grad().

  saschafrey asked Jul 21, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  lowering / cost analysis of @nnx.jit functions

  lzanini asked Jul 19, 2024 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Flax Quick Start Tutorial requires 7GB of VRAM ?! help please!

  yCobanoglu asked Jul 13, 2024 in Q&A · Closed · Unanswered

  

  

  1
• 2 You must be logged in to vote

  🙌

  FlaxDiff: A Diffusion library but with simple code and good explainations

  AshishKumar4 asked Jul 8, 2024 in Show and tell · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  nnx using vmap to create multiple models

  JeyRunner asked Jul 2, 2024 in Q&A · Answered

  

  

  4