profile
viewpoint

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

https://github.com/google/jax

google

viewpoint

Express your opinions freely and help others including your future self

Issues rank

GPU not detected by JAX. Warning: No GPU/TPU found, falling back to CPU. hot 43
Could not load dynamic library 'libcudnn.so.8' hot 34
TypeError: Gradient only defined for scalar-output functions. Output had shape: (1,) hot 25
how to detect if GPU is being used? (feature request) hot 16
Installing from source using Conda and CUDA could be improved - jax hot 12
Tips for debugging NaNs in gradient? hot 11
Clear GPU memory hot 11
Jax on WSL with CUDA gives me error: "could not open file to read NUMA node" hot 11
Possible reasons for GPU not being found? hot 10
TypeError: Gradient only defined for scalar-output functions. Output had shape: (1,) hot 10
cannot find libdevice hot 10
no kernel image is available for execution on the device hot 9
Unimplemented: DNN library is not found. hot 9
Custom VJPs for external functions hot 8
cuSolver internal error during QR decomposition on GPU hot 7

contributors (According to the first 100)

mattjj
hawkinsp
jakevdp
gnecula
froystig
skye
apaszke
shoyer
bchetioui
jekbradbury
jblespiau
zhangqiaorjc
fehiepsi
j-towns
yashk2810
dougalm
levskaya
sharadmv
tomhennigan
LenaMartens
juliuskunze
ekelsen
alexbw
jacobjinkelly
majnemer
minoring
dpfau
wyjw
chr1sj0nes
marcvanzee
aslanides
alexalemi
inailuig
colemanliyah
pschuh
trevorcai
vishwakftw
romanngg
HerculesJack
danieldjohnson
joaogui1
sschoenholz
TinaZetongJia
mtthss
8bitmp3
lgeiger
qiuminxu
dsweaver
cloudhan
clemisch
gsp-27
kroq-gar78
aeftimia
NeilGirdhar
superbobry
ayush-1506
rsepassi
xysun
nicholasjng
petebu
adarob
brianwa84
samuela
gmittal
jonasrauber
MikeInnes
srvasude
agrue
cheshire
Helw150
mkunesch
reza-amd
TuanNguyen27
aldragan0
deniskamazur
PhilipVinc
JasperSnoek
terhorst
jlebar
lucasb-eyer
lukepfister
n2cholas
proteneer
zhongwen
IgorWilbert
yurodiviy
zafarali
duvenaud
navneet-nmk
botev
brettkoonce
johnpjf
Bharat123rox
SiegeLordEx
jheek
alexminnaar
ahoenselaar
wdphy16
fabianp
gdahia