Open Issues Need Help
View All on GitHub improve pjit pytree error involving Nones (and PytreeLeafs) about 2 months ago
good first issue better_errors
jax-ml/jax
33.4K
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python
#jax
Support profiling without source code changes 2 months ago
enhancement good first issue
jax-ml/jax
33.4K
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python
#jax
lax.conv_transpose has different convention than TF, document it and make switching easy? 2 months ago
enhancement good first issue documentation
jax-ml/jax
33.4K
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python
#jax