Skip to content

Pull requests: jax-ml/jax

New pull request New
403 Open 15,848 Closed
403 Open 15,848 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Stop using generators for linear_util transformations pull ready Ready for copybara import and testing
#24873 opened Nov 13, 2024 by dougalm Loading…
1
Add numpy.put_along_axis.
#24871 opened Nov 12, 2024 by carlosgmartin Loading…
1
@jakevdp
9
finalize deprecation of jax.clear_backends pull ready Ready for copybara import and testing
#24865 opened Nov 12, 2024 by jakevdp Loading…
@jakevdp
jnp.logaddexp2: simplify implementation pull ready Ready for copybara import and testing
#24864 opened Nov 12, 2024 by jakevdp Loading…
@jakevdp
Move the Control Flow section of Sharp Bits into its own tutorial. documentation
#24862 opened Nov 12, 2024 by emilyfertig Loading…
@jakevdp
9
[JAX/version] Support adding custom version suffix.
#24861 opened Nov 12, 2024 by yliu120 Loading…
2
[JAX/distributed] Small clean ups on warning emissions in jax.distributed.
#24853 opened Nov 12, 2024 by yliu120 Loading…
Bump hypothesis from 6.102.4 to 6.118.7 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24840 opened Nov 11, 2024 by dependabot bot Loading…
Add Pallas Triton lowering for jax.lax.bitcast_convert_type.
#24834 opened Nov 11, 2024 by copybara-service bot Loading…
@petebu
Improvements in the handling of tracing debug info pull ready Ready for copybara import and testing
#24831 opened Nov 11, 2024 by gnecula • Draft
@gnecula
Pattern match dot algorithm spec to preset name pull ready Ready for copybara import and testing
#24820 opened Nov 9, 2024 by dfm Loading…
@dfm
Adds coverage for spmd-axisname-filtering in shard_map transpose.
#24811 opened Nov 8, 2024 by copybara-service bot Loading…
Specialize ufunc.reduce for monoidal binary ufuncs. pull ready Ready for copybara import and testing
#24810 opened Nov 8, 2024 by jakevdp Loading…
@jakevdp
2
[Mosaic GPU] Add CUPTI profiler alongside events-based implementation
#24805 opened Nov 8, 2024 by andportnoy Loading…
[AutoPGLE] Explicitly ignore host callback pointers
#24797 opened Nov 8, 2024 by copybara-service bot Loading…
Check mem_stats key in test_lax_full_like_efficient
#24796 opened Nov 8, 2024 by apivovarov Loading…
Fix parallel pgle-tests execution.
#24793 opened Nov 8, 2024 by copybara-service bot Loading…
Add new CI scripts for running Pytests
#24790 opened Nov 8, 2024 by copybara-service bot Loading…
Add commands to run Bazel GPU (non-RBE) jobs
#24788 opened Nov 8, 2024 by copybara-service bot Loading…
Add new CI scripts for building JAX artifacts
#24785 opened Nov 8, 2024 by copybara-service bot Loading…
[pallas] Allow user to pass 64-bit indices to pl.{load,store,...}.
#24782 opened Nov 8, 2024 by copybara-service bot Loading…
@chr1sj0nes
[Mosaic TPU] Support 1D concat: set implicit_dim to kSecondMinor to treat 1D (N,) as (1, N) and then tile it as (1, 128)
#24781 opened Nov 8, 2024 by copybara-service bot Loading…
[mgpu] Broadcast the LHS fragmented array if it is splat.
#24780 opened Nov 8, 2024 by copybara-service bot Loading…
[Pallas] Increase test coverage of pl.dot.
#24777 opened Nov 8, 2024 by copybara-service bot Loading…
Won't submit. Experimenting.
#24766 opened Nov 7, 2024 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.