Skip to content

Pull requests: jax-ml/jax

New pull request New
548 Open 21,844 Closed
548 Open 21,844 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
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add caveat about cond and vmap to example
#27203 opened Mar 17, 2025 by alik-git Loading…
jax.nn.initializers: use Sequence[int] directly for shape annotations. kokoro:force-run pull ready Ready for copybara import and testing
#27202 opened Mar 17, 2025 by jakevdp Loading…
1
@jakevdp
1
Update cpu_feature_guard.c
#27125 opened Mar 13, 2025 by 87dnz Loading…
1
Update Controlling data and computation placement on devices section in FAQs
#27085 opened Mar 12, 2025 by rajasekharporeddy Loading…
1
1
Add a custom_partitioning helper to support batch partitioning pull ready Ready for copybara import and testing
#27067 opened Mar 11, 2025 by dfm Loading…
@dfm
replaced np.array with np.fromfile for performance improvements
#27018 opened Mar 9, 2025 by ParamThakkar123 Loading…
@jakevdp
1
[DOC] Add documentation for jax_debug_infs configuration option to JAX debugging flags
#27012 opened Mar 8, 2025 by sumana-2705 Loading…
2
[Mosaic GPU] Check in WIP grouped GEMM
#26997 opened Mar 7, 2025 by andportnoy • Draft
@apaszke
11
jax.random.uniform: add resolution='high' argument
#26974 opened Mar 6, 2025 by jakevdp Loading…
@jakevdp
Update remat config name in docs
#26929 opened Mar 5, 2025 by danijar Loading…
Retry connection errors to combat flakiness in google metadata server during tpu initialization (reopens #24432)
#26919 opened Mar 4, 2025 by dlwh Loading…
2
jax profiler - trace events with absolute timestamps #26288
#26873 opened Mar 1, 2025 by Utkarsh-Aggarwal Loading…
1
Add plugin versions to print_environment_info.
#26768 opened Feb 26, 2025 by dfm Loading…
@dfm
Replace setup.py with pyproject.toml
#26683 opened Feb 23, 2025 by NeilGirdhar Loading…
@hawkinsp
4
docs: Add note about computation perturbation in debug.print/breakpoint
#26682 opened Feb 22, 2025 by alhridoy Loading…
1
8
[pallas:triton] Fix atomic min/max lowering for uint and float types pull ready Ready for copybara import and testing
#26604 opened Feb 19, 2025 by draganmladjenovic Loading…
5
[better_errors] Simplify api_util.debug_info. pull ready Ready for copybara import and testing
#26603 opened Feb 19, 2025 by gnecula Loading…
@gnecula
6
[Will not merge] bug report on autodiff on partial auto nested shard_map
#26588 opened Feb 18, 2025 by yliu120 • Draft
1
feat(gh-13291): Add Pareto distribution functions: cdf, logcdf, logsf, sf, ppf, logppf, and corresponding tests
#26556 opened Feb 15, 2025 by Qazalbash Loading…
Codebeaver/test 723699680 26477
#26492 opened Feb 12, 2025 by CodeBeaverAI Loading…
1
Make jnp.asarray lower to asarray_p in the simplest cases.
#26244 opened Jan 31, 2025 by jakevdp • Draft
@jakevdp
[mutable-arrays] linearize the state
#26176 opened Jan 29, 2025 by mattjj Loading…
Update JVP rule for abs to fix behavior for complex infinite inputs pull ready Ready for copybara import and testing
#26086 opened Jan 24, 2025 by dfm Loading…
@dfm
11
Implement lax.full_like in terms of a primitive pull ready Ready for copybara import and testing
#26045 opened Jan 22, 2025 by jakevdp Loading…
@jakevdp
Remove linear_util.cache pull ready Ready for copybara import and testing
#26038 opened Jan 22, 2025 by dougalm Loading…
ProTip! Adding no:label will show everything without a label.