Discussions

• 1 You must be logged in to vote

  🙏

  vmapping lambda functions

  aadityacs asked Sep 19, 2024 in Q&A · Answered

  

  

  

  

  4
• 2 You must be logged in to vote

  🙏

  How can I run part of jit compiled function on a different device?

  n-gao asked Dec 3, 2023 in Q&A · Unanswered

  

  

  

  

  8
• 1 You must be logged in to vote

  🙏

  Best way compute dynamic slices of arrays where the slices are known beforehand?

  Mattias421 asked Sep 13, 2024 in Q&A · Unanswered

  

  

  4
• 1 You must be logged in to vote

  🙏

  What is stop_gradient used for?

  cool-RR asked Sep 17, 2024 in Q&A · Closed · Answered

  

  

  7
• 1 You must be logged in to vote

  🙏

  Branching behaviour depending on device used.

  AdrienCorenflos asked Sep 18, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Producing custom array sharding

  luyug asked Sep 17, 2024 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Recommended approach for forward mode value_and_grad

  lockwo asked Sep 15, 2024 in Q&A · Unanswered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Running multiple jobs, one per TPU core?

  rog77 asked Jul 5, 2023 in Q&A · Answered

  

  

  

  

  6
• 1 You must be logged in to vote

  #️⃣

  jax._src.core.check_scalar_conversion

  lfaucheux asked Sep 16, 2024 in General · Answered

  

  

  1
• 1 You must be logged in to vote

  #️⃣

  Clarification regarding GPU performance

  AakashKumarNain asked Sep 13, 2024 in General · Unanswered

  

  

  

  

  3
• 1 You must be logged in to vote

  🙏

  Optimizer state for pre-trained model

  VinaySingh561 asked Sep 13, 2024 in Q&A · Unanswered

  

  

  4
• 1 You must be logged in to vote

  🙏

  How to see what's new for each version？

  south-ocean asked Sep 14, 2024 in Q&A · Answered

  

  

  1
• 1 You must be logged in to vote

  🙏

  How do I do this rather trivial parallelization?

  sokol11 asked Sep 13, 2024 in Q&A · Answered

  

  

  13
• 1 You must be logged in to vote

  🙏

  How to compile the jaxlib with debug symbols?

  juuso-oskari asked Sep 10, 2024 in Q&A · Unanswered

  

  

  2
• 3 You must be logged in to vote

  🙏

  The NVIDIA driver's CUDA version is 12.2 which is older than the ptxas CUDA version (12.3.107). Because the driver is older than the ptxas version, XLA is disabling parallel compilation, which may slow down compilation. You should update your NVIDIA driver or use the NVIDIA-provided CUDA forward compatibility packages.

  VachanVY asked Mar 10, 2024 in Q&A · Unanswered

  

  

  

  

  22
• 1 You must be logged in to vote

  🙏

  GPU not detected in Docker

  rdwight asked Sep 10, 2024 in Q&A · Answered

  

  

  

  

  4
• 20 You must be logged in to vote

  📢

  moving our GitHub repository to jax-ml

  froystig announced Aug 29, 2024 in Announcements

  

  

  1
• 1 You must be logged in to vote

  🙏

  Computing gradients involving sparse arrays

  SNMS95 asked Aug 9, 2024 in Q&A · Answered

  

  

  5
• 1 You must be logged in to vote

  🙏

  Methods Efficiently Finding Unique Rows in a 2D Integer Array Without Using jnp.unique Along Axis?

  AndrewLiu0725 asked Sep 10, 2024 in Q&A · Unanswered

  

  

  3
• 1 You must be logged in to vote

  🙏

  Jax sparse-sparse matrix multiplication memory usage

  ivirshup asked Aug 23, 2023 in Q&A · Answered

  

  

  

  

  18
• 5 You must be logged in to vote

  🙏

  How to build debug-version jaxlib?

  ronghongbo asked Mar 7, 2022 in Q&A · Unanswered

  

  

  

  

  2
• 1 You must be logged in to vote

  🙏

  Export stablehlo from jax function with kwargs

  sechkova asked Sep 9, 2024 in Q&A · Unanswered

  

  

  2
• 1 You must be logged in to vote

  #️⃣

  Why does SVD no longer determine whether the matrix contains infinite numbers?

  mayuyuace asked Aug 26, 2024 in General · Closed · Answered

  

  

  

  

  28
• 1 You must be logged in to vote

  💡

  Making scan() and cond() functions more usable

  MRiabov asked Sep 7, 2024 in Ideas · Unanswered

  

  

  7
• 1 You must be logged in to vote

  🙏

  Jax/Flax models 2x slower on Sapphire Rapids (c7i) than Ice Lake (c6i) instances | x86

  Rohanjames1997 asked Aug 28, 2024 in Q&A · Unanswered

  

  

  

  

  10