juliuskunze/jaxnet

Issues

• Issues with basic RNN...

  #26 opened 4 years ago by adarob
  1

• ImportError: cannot import name 'unzip2' from 'jax'

  #25 opened 4 years ago by asmith26
  2

• PRNG handling akin to parameters

  #24 opened 5 years ago by samuela
  2

• Getting RuntimeError in resnet50 example

  #22 opened 5 years ago by harpone
  1

• Automate random key injection (simplify dropout etc)

  #21 opened 5 years ago by juliuskunze
  0

• There are some memory leak

  #7 opened 5 years ago by neon5d
  2

• Memory usage is too high

  #11 opened 5 years ago by juliuskunze
  1

• Fix PixelCNN's weight norm layers

  #17 opened 5 years ago by juliuskunze
  0

• Allow vmap(parametrized(fun))

  #16 opened 5 years ago by juliuskunze
  1

• Add Batched(parametrized)

  #20 opened 5 years ago by juliuskunze
  0

• Feature Request: Add Mish Activation

  #19 opened 5 years ago by digantamisra98
  2

• Allow parametrized(jit(fun))

  #14 opened 5 years ago by juliuskunze
  0

• Cache miss breaks jit(parametrized(jit).apply)

  #18 opened 5 years ago by juliuskunze
  0

• Allow lax.scan with unparametrized cell within parametrized function

  #15 opened 5 years ago by juliuskunze
  0

• Repair unbatched PixelCNN example

  #5 opened 5 years ago by juliuskunze
  0

• Fix parameter sharing between multiple parents

  #8 opened 5 years ago by juliuskunze
  0

• Fix tuple/dict inputs

  #9 opened 5 years ago by juliuskunze
  0

• Remove need for input-dependent dummy argument

  #1 opened 5 years ago by juliuskunze
  0

• Module parsing is slow

  #10 opened 5 years ago by juliuskunze
  0

• RNN training fails

  #2 opened 5 years ago by juliuskunze
  0

• Allow overlap when reusing parameters

  #13 opened 5 years ago by juliuskunze
  0

• Repair RNNs

  #4 opened 5 years ago by juliuskunze
  0

• Allow storing optimizer state

  #12 opened 5 years ago by juliuskunze
  0

• os.environ["CUDA_VISIBLE_DEVICES"] not working

  #6 opened 5 years ago by neon5d
  1

• Update to jax version 0.1.42

  #3 opened 5 years ago by juliuskunze
  0