Skip to content

/ jax

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

520 Open 1,138 Closed
520 Open 1,138 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels.
Projects
Filter by project
Milestones
Filter by milestone
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Computing forward gradients in a compressed manner.
#4516 opened Oct 8, 2020 by eneftci
cuSolver internal error during QR decomposition on GPU
#4497 opened Oct 8, 2020 by janosg
Huge difference in speed between first and subsequent runs
#4495 opened Oct 8, 2020 by gnool
1
Building on linux with ppc64le CPU contributions welcome
#4493 opened Oct 8, 2020 by f0uriest
2
np.mean raises warning when input is float64 in jax 0.2.1 better_errors
#4490 opened Oct 8, 2020 by fehiepsi
@jakevdp
1
Embedded GPUs and TPUs with JAX
#4487 opened Oct 7, 2020 by aizzaac
2
jax.numpy.asarray from a numpy.ndarray should not copy the data for CPU backend (and vice vesa)
#4486 opened Oct 7, 2020 by zhou13
installation error: No library found under: /usr/local/cuda-11.1/targets/x86_64-linux/lib/libcudart.so.11.1
#4481 opened Oct 7, 2020 by rezah
3
Cross multiplication on JAX is faster in CPU compared to GPU question
#4474 opened Oct 7, 2020 by CleanPegasus
1
Support for variable length in jitted scans? question
#4473 opened Oct 7, 2020 by nrontsis
@froystig
1
Training larger model is (much) faster than smaller model
#4472 opened Oct 7, 2020 by jli05
5
Integer matrix multiplication with int8 input and int32 accumulation enhancement
#4465 opened Oct 7, 2020 by malmaud
1
Add Python 3.9 wheels for jaxlib enhancement
#4461 opened Oct 6, 2020 by NeilGirdhar
2
conda installation cuda incompatibility error invalid
#4450 opened Oct 4, 2020 by andyorange
1
Haiku `avg_pool` results in assertion error during backwards pass bug
#4435 opened Oct 1, 2020 by sharadmv
2
jnp.zeros does not support float0 as a dtype
#4433 opened Oct 1, 2020 by danieldjohnson
1
1
No tools in profiler
#4427 opened Sep 30, 2020 by tancik
4
bfloat16 error from NumPy when combining TensorFlow + JAX
#4426 opened Sep 30, 2020 by shoyer
1
Unclear shape mismatch error from using jax.mask bug
#4420 opened Sep 29, 2020 by phinate
1
New `TypeError: <class 'function'> is not a valid JAX type` exception in JAX 0.2.0 question
#4416 opened Sep 29, 2020 by rlouf
@mattjj
1
Feature request: support more flexible handling of seeds and seed-like objects like `numpy.random.SeedSequence`
#4399 opened Sep 24, 2020 by gehring
2
Help needed for creating a package for gentoo
#4396 opened Sep 23, 2020 by epsilon-0
Warn when there are too many recompilations better_errors enhancement
#4390 opened Sep 23, 2020 by gnecula
3
grad transformed functions should not return NumPy arrays
#4386 opened Sep 23, 2020 by shoyer
Benchmarking Memory Taken By Neural Network question
#4385 opened Sep 23, 2020 by pranavsubramani
@mattjj
3
ProTip! Adding no:label will show everything without a label.
You can’t perform that action at this time.