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.

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

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

Issues list

Improvements to debug_nans enhancement New feature or request
#25643 opened Dec 20, 2024 by emilyfertig
5 tasks
Add mask argument to lax.argmax enhancement New feature or request
#25623 opened Dec 20, 2024 by carlosgmartin
error: 'string' in namespace 'std' does not name a type bug Something isn't working
#25598 opened Dec 19, 2024 by adamjstewart
Make xla_bridge.is_gpu more extensible enhancement New feature or request
#25521 opened Dec 17, 2024 by PragmaTwice
Public API for mesh axis index size enhancement New feature or request
#25515 opened Dec 16, 2024 by gabbard
jax.random.choice(replace=True) samples 0 probability index bug Something isn't working duplicate This issue or pull request already exists
#25498 opened Dec 16, 2024 by LemonATsu
register_dataclass does not handle __init__ methods bug Something isn't working
#25486 opened Dec 13, 2024 by patrick-kidger
Slow convolution, many memory warnings bug Something isn't working
#25461 opened Dec 13, 2024 by apasarkar
Segmentation fault in jaxlib 0.4.37 and Debian 11 ARM64 bug Something isn't working
#25436 opened Dec 12, 2024 by gongomgra
Latency Hiding Scheduler not working with jax 0.4.35 bug Something isn't working
#25404 opened Dec 11, 2024 by lgiacomoni
jax.full allocates memory on the wrong device bug Something isn't working
#25396 opened Dec 11, 2024 by Chaosruler972
Error when donating buffer with under-specified sharding bug Something isn't working
#25346 opened Dec 9, 2024 by jaro-sevcik
[Pallas] Add lowering support for tl.join enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#25321 opened Dec 6, 2024 by ggengnv
ProTip! Find all open issues with in progress development work with linked:pr.