juliuskunze jaxnet issues

juliuskunze / jaxnet

Concise deep learning for JAX

Apache License 2.0

184 stars 14 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

Issues with basic RNN...

#26 adarob closed 4 years ago
1
ImportError: cannot import name 'unzip2' from 'jax'

#25 asmith26 closed 4 years ago
2
PRNG handling akin to parameters

#24 samuela opened 4 years ago
2
Infer nr_mix in pcnn loss instead of hard-coding it

#23 j-towns closed 4 years ago
1
Getting RuntimeError in resnet50 example

#22 harpone closed 5 years ago
1
Automate random key injection (simplify dropout etc)

#21 juliuskunze closed 5 years ago
0
Add Batched(parametrized)

#20 juliuskunze closed 5 years ago
0
Feature Request: Add Mish Activation

#19 digantamisra98 closed 5 years ago
2
Cache miss breaks jit(parametrized(jit).apply)

#18 juliuskunze opened 5 years ago
0
Fix PixelCNN's weight norm layers

#17 juliuskunze closed 5 years ago
0
Allow vmap(parametrized(fun))

#16 juliuskunze closed 5 years ago
1
Allow lax.scan with unparametrized cell within parametrized function

#15 juliuskunze closed 5 years ago
0
Allow parametrized(jit(fun))

#14 juliuskunze closed 5 years ago
0
Allow overlap when reusing parameters

#13 juliuskunze closed 5 years ago
0
Allow storing optimizer state

#12 juliuskunze closed 5 years ago
0
Memory usage is too high

#11 juliuskunze closed 5 years ago
1
Module parsing is slow

#10 juliuskunze closed 5 years ago
0
Fix tuple/dict inputs

#9 juliuskunze closed 5 years ago
0
Fix parameter sharing between multiple parents

#8 juliuskunze closed 5 years ago
0
There are some memory leak

#7 neon5d closed 5 years ago
2
os.environ["CUDA_VISIBLE_DEVICES"] not working

#6 neon5d closed 5 years ago
1
Repair unbatched PixelCNN example

#5 juliuskunze closed 5 years ago
0
Repair RNNs

#4 juliuskunze closed 5 years ago
0
Update to jax version 0.1.42

#3 juliuskunze closed 5 years ago
0
RNN training fails

#2 juliuskunze closed 5 years ago
0
Remove need for input-dependent dummy argument

#1 juliuskunze closed 5 years ago
0