Skip to content

Issues: jax-ml/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.

Sign up for GitHub

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

1,415 Open 4,247 Closed
1,415 Open 4,247 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

JAX Profiling in Colab contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. documentation P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#3694 opened Jul 8, 2020 by 8bitmp3
@8bitmp3
9
Clarification on which objects are multiprocess/fork safe?
#3691 opened Jul 8, 2020 by proteneer
2
BDF solver for stiff ODEs
#3686 opened Jul 7, 2020 by skrsna
5
scipy.ndimage functions?
#3647 opened Jul 3, 2020 by benjaminpope
11
jnp.einsum does not catch shape mismatch when optimize=True
#3645 opened Jul 2, 2020 by EddieCunningham
7
Add option to have multiple custom gradients for one function enhancement New feature or request
#3641 opened Jul 2, 2020 by dpfau
5
Unexpected overhead when using grad(pmap) performance make things lean and fast
#3612 opened Jun 30, 2020 by izmailovpavel
Issue with JIT when compiling scan function with given length better_errors Improve the error reporting question Questions for the JAX team
#3603 opened Jun 30, 2020 by markusschmitt
3
Using iterator with fori_loop documentation question Questions for the JAX team
#3567 opened Jun 26, 2020 by nissy-dev
@mattjj
5
Longer docstring for jax.jit? documentation question Questions for the JAX team
#3514 opened Jun 22, 2020 by juesato
@mattjj
4
Update Jax Quickstart for pmap, remat documentation
#3458 opened Jun 16, 2020 by 8bitmp3
4
odeint Runge-Kutta step size sometimes goes backward in time bug Something isn't working
#3437 opened Jun 14, 2020 by samuela
2
6
Initial benchmarking of jax on some simple kernels performance make things lean and fast
#3411 opened Jun 11, 2020 by proteneer
2
pmap sometimes fails under disable_jit() with jax.core.ConcretizationTypeError bug Something isn't working
#3401 opened Jun 11, 2020 by Joshuaalbert
2
jax.numpy.reshape does not convert result to a DeviceArray
#3391 opened Jun 9, 2020 by brianwa84
5
programmatic profiling enhancement New feature or request
#3374 opened Jun 9, 2020 by mts42000
5
Question: efficient gradient computation for multi-linear function application question Questions for the JAX team
#3366 opened Jun 8, 2020 by Jakob-Unfried
16
Individual hessian contributions of params used in a scanned function question Questions for the JAX team
#3353 opened Jun 7, 2020 by Victorlouisdg
Dynamically sized arrays with maximum length enhancement New feature or request question Questions for the JAX team
#3345 opened Jun 6, 2020 by Justin-Tan
@mattjj
8
Choice of ops for stencil computations run on TPUs question Questions for the JAX team
#3341 opened Jun 5, 2020 by hubertlu-tw
1
Type checking for pytrees
#3340 opened Jun 5, 2020 by shoyer
16
bfloat16/float32 memory requirements seem off bug Something isn't working
#3302 opened Jun 2, 2020 by btanner
3
vmapping list appends and performance question Questions for the JAX team
#3297 opened Jun 2, 2020 by choltz95
@mattjj
2
Buffer donation optimizations in AD enhancement New feature or request
#3286 opened Jun 2, 2020 by jekbradbury
Hoist large constants into parameters enhancement New feature or request
#3220 opened May 27, 2020 by hawkinsp
@froystig
6
ProTip! Mix and match filters to narrow down what you’re looking for.