Discussions

• 1 You must be logged in to vote

  🙏

  Parameters not available as instance properties after setup()

  ingmarschuster asked Feb 9, 2021 in Q&A · Answered

  

  

  3
• 1 You must be logged in to vote

  🙏

  Derivative of output with respect to input

  lweitkamp asked Feb 9, 2021 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Creating a running average with self.variable

  matthias-wright asked Feb 7, 2021 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Trouble understading error message : ValueError: Can't call compact methods on unbound modules

  LysSanzMoreta asked Feb 4, 2021 in Q&A · Unanswered

  

  

  4
• 1 You must be logged in to vote

  🙏

  How to swap in a new variable tree for a given collection in a submodule in Flax/JAX?

  marcvanzee asked Feb 5, 2021 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Scalar parameters with constraints

  ingmarschuster asked Feb 4, 2021 in Q&A · Answered

  

  

  4
• 1 You must be logged in to vote

  🙏

  How to assign methods as Flax Module attributes?

  marcvanzee asked Feb 5, 2021 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  How to make an Optax optimizer in Flax?

  marcvanzee asked Feb 5, 2021 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  What are variable dicts, variable collections, and the mutable argument to Module.apply in Flax?

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

  avital asked Dec 29, 2020 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Is there a 2D "masked convolution" in JAX/Flax?

  avital asked Feb 3, 2021 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  How flax classes are aware of their class variables?

  MiladInk asked Feb 3, 2021 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  #️⃣

  Jax transforms and modules cannot be mixed.

  yang-song asked Feb 1, 2021 in General · Answered

  

  

  

  

  5
• 1 You must be logged in to vote

  💡

  💡 README: Improve discoverability of Flax Basics 🔍

  8bitmp3 asked Dec 20, 2020 in Ideas · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  Creating module attributes with @nn.compact

  matthias-wright asked Jan 24, 2021 in Q&A · Answered

  

  

  

  

  4
• 1 You must be logged in to vote

  🙏

  How to use bucketing for a dataset in a multi-host setting

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

  

  

  1
• 1 You must be logged in to vote

  🙏

  How to use shape-dependent variables without @nn.compact in JAX/Flax?

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

  

  

  1
• 1 You must be logged in to vote

  🙏

  How to do Batch Normalization in JAX/Flax?

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

  

  

  1
• 2 You must be logged in to vote

  🙏

  What is the difference between Module.param() and Module.variable()?

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

  

  

  1
• 1 You must be logged in to vote

  🙏

  What is the Flax equivalent of Tensorflow functions?

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

  

  

  2
• 1 You must be logged in to vote

  🙏

  How to implement beam search in JAX/Flax?

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

  

  

  1
• 1 You must be logged in to vote

  #️⃣

  Transfer Learning and variable collections

  cgarciae asked Jan 15, 2021 in General · Unanswered

  

  

  

  

  4
• 2 You must be logged in to vote

  🙏

  How to calculate the CTC Loss by jax and flax?

  Du-Yao asked Jan 15, 2021 in Q&A · Unanswered

  

  

  1
• 1 You must be logged in to vote

  🙌

  👍 Flax and JAX in Google AI Blog

  8bitmp3 asked Jan 14, 2021 in Show and tell · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Using jax.mask with BiRNN for text classification

  myagues asked Dec 13, 2020 in Q&A · Unanswered

  

  

  

  

  6
• 1 You must be logged in to vote

  🙏

  Use of jax.mask in seq2seq linen example

  iislucas asked Jan 8, 2021 in Q&A · Answered

  

  

  2