Discussions

• 1 You must be logged in to vote

  🙏

  Incompatible CUBIN for Distributed primitive

  ASKabalan asked Apr 27, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Help to speed up code

  JohannesEsslinger asked Apr 25, 2024 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Slice and repeat

  AakashKumarNain asked Apr 26, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  JAX error in TPU v2-8 vs v2-32

  gpkmohan asked Apr 25, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  jax to tflite conversion

  ASEM000 asked Apr 19, 2024 in Q&A · Answered

  

  

  

  

  4
• 1 You must be logged in to vote

  💡

  Can Jax generate numerically stable functions?

  NeilGirdhar asked Apr 24, 2024 in Ideas · Unanswered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Using jax.pmap/vmap and jax.lax.switch to execute a simulation in parallel. Is it an anti-pattern?

  SimiPixel asked Apr 24, 2024 in Q&A · Unanswered

  

  

  0
• 7 You must be logged in to vote

  🙏

  Seeking optimization advice for interpolation-heavy computation

  Qazalbash asked Apr 23, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Seek help understanding the profile trace of a GPU jit execution

  jjyyxx asked Apr 23, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Batched index update of sharded arrays without communication?

  inailuig asked Apr 23, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Error occurred while evaluating genome 2: JAX encountered invalid PRNG key data: expected key_data.ndim >= 1; got ndim=0

  kehsan asked Apr 16, 2024 in Q&A · Unanswered

  

  

  13
• 1 You must be logged in to vote

  🙏

  how does shard_map work with scan?

  jpilaul asked Apr 19, 2024 in Q&A · Answered

  

  

  5
• 1 You must be logged in to vote

  🙏

  jax.Array.view returns a copy rather than a view

  hirayaku asked Apr 22, 2024 in Q&A · Answered

  

  

  3
• 2 You must be logged in to vote

  🙏

  Persistent Compilation Cache - explicit storing inner function

  JeyRunner asked Apr 17, 2024 in Q&A · Unanswered

  

  

  4
• 2 You must be logged in to vote

  🙏

  How to deal with impure method with side-effect for registered Pytree class?

  HGangloff asked Apr 19, 2024 in Q&A · Answered

  

  

  

  

  6
• 1 You must be logged in to vote

  🙏

  In-place update for memory efficient behavior

  sh0416 asked Apr 22, 2024 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Compare the equality of address between two arrays.

  sh0416 asked Apr 20, 2024 in Q&A · Answered

  

  

  3
• 1 You must be logged in to vote

  🙏

  What to expect from Offloading in Remat?

  luyug asked Dec 20, 2023 in Q&A · Unanswered

  

  

  

  

  8
• 1 You must be logged in to vote

  🙏

  Are there jit compile-time guards?

  tillahoffmann asked Apr 12, 2024 in Q&A · Answered

  

  

  

  

  7
• 2 You must be logged in to vote

  #️⃣

  Challenges porting gaussian splatting to Jax

  peabody124 asked Dec 26, 2023 in General · Unanswered

  

  

  

  

  3
• 9 You must be logged in to vote

  📢

  changes coming to JAX's RNG – auto-parallelizable by default

  froystig announced Nov 10, 2023 in Announcements

  

  

  

  

  4
• 1 You must be logged in to vote

  🙏

  help with categorical encoding

  g00phy asked Apr 18, 2024 in Q&A · Unanswered

  

  

  2
• 1 You must be logged in to vote

  💡

  Implementing a FID Curve with diffusers

  EvergreenTree asked Apr 18, 2024 in Ideas · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  How to perform a gridsearch over multiple models in parallel with hyperparameters that can change in vmap?

  kclauw asked Apr 15, 2024 in Q&A · Unanswered

  

  

  1
• 3 You must be logged in to vote

  #️⃣

  Does compilation_cache not work for gpus and cpus?

  OrenLeung asked Dec 20, 2022 in General · Answered

  

  

  

  

  11