Issues

• add unfold and extract_image_patches wrappers around conv_general_dilated_patches

  enhancementopenP2 (eventual)
  jax-ml/jax#6594 · Filter by author mattjj opened on Apr 29, 2021
  1

• Do not copy buffers that are unmodified

  enhancementopen
  jax-ml/jax#6284 · Filter by author dionhaefner opened on Mar 30, 2021
  1

• scipy.linalg.solve_sylvester

  enhancementopen
  jax-ml/jax#6089 · Filter by author shoyer opened on Mar 16, 2021
  2

• map_coordinates mode='constant' not properly applied

  bugdocumentationopen
  jax-ml/jax#5687 · Filter by author agudallago opened on Feb 9, 2021
  6

  

• Slow jit with list argument

  bugopen
  jax-ml/jax#5684 · Filter by author tetterl opened on Feb 9, 2021
  3

  

• Is it possible to have early stopping in lax.scan?

  enhancementopen
  jax-ml/jax#5642 · Filter by author salayatana66 opened on Feb 5, 2021
  7

• Function with static_argnums as jited function argument.

  enhancementopen
  jax-ml/jax#5609 · Filter by author jackd opened on Feb 3, 2021
  7

• static_argnums for parts of the pytree object

  contributions welcomeenhancementopen
  jax-ml/jax#5487 · Filter by author PgLoLo opened on Jan 21, 2021
  2

• Implement jax.scipy.linalg.logm

  contributions welcomeenhancementopen
  jax-ml/jax#5469 · Filter by author jaeyoo opened on Jan 20, 2021
  18

• [FR] Support for scipy.special.gammaincinv

  enhancementopen
  jax-ml/jax#5350 · Filter by author fehiepsi opened on Jan 8, 2021
  11

• Jax convolution significantly slower than scipy

  contributions welcomeCPUenhancementNVIDIA GPUopenP1 (soon)XLA
  jax-ml/jax#5227 · Filter by author smartalecH opened on Dec 18, 2020
  8

• Spoof multiple hosts

  enhancementopen
  jax-ml/jax#5155 · Filter by author alewis opened on Dec 10, 2020
  15

• Making function vmap'able

  open
  jax-ml/jax#4157 · Filter by author renos opened on Aug 26, 2020
  3

• Scaling behaviour of jax.test_util.check_grads

  bugopen
  jax-ml/jax#3130 · Filter by author nsavinov opened on May 18, 2020
  7