Skip to content

Pull requests: jax-ml/jax

New pull request New
563 Open 21,898 Closed
563 Open 21,898 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

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…
Implement gaussian_filter1d and gaussian_filter
#26011 opened Jan 21, 2025 by adonath Loading…
2 tasks
@jakevdp
2
Clarify type hints for wrapped functions pull ready Ready for copybara import and testing
#25994 opened Jan 21, 2025 by ncoghlan Loading…
@superbobry
5
fix aliased arg in async pallas docs
#25977 opened Jan 18, 2025 by arvoelke Loading…
[Will not merge] Show the IR of the partial-auto boundary.
#25813 opened Jan 9, 2025 by yliu120 Loading…
12
New Experimental JAX Check Numerics API.
#25785 opened Jan 8, 2025 by epiqueras Loading…
@froystig @mattjj
2
[Shmap/PartialAuto] Temporary solution for debug.print inside a partial-auto shard map.
#25705 opened Jan 1, 2025 by yliu120 Loading…
1
WIP: no special treatment for ShapeDtypeStruct
#25583 opened Dec 18, 2024 by jakevdp • Draft
@jakevdp
1
Fix debug_nans regressions. pull ready Ready for copybara import and testing
#25519 opened Dec 16, 2024 by emilyfertig Loading…
1
8
Bump fonttools from 4.51.0 to 4.55.3 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25512 opened Dec 16, 2024 by dependabot bot Loading…
Add a workflow to run benchmarks
#25386 opened Dec 10, 2024 by juliagmt-google Loading…
1
ProTip! Type g i on any issue or pull request to go back to the issue listing page.