Skip to content

Pull requests: jax-ml/jax

New pull request New
428 Open 16,022 Closed
428 Open 16,022 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

Add float8_e8m0fnu type support
#25116 opened Nov 26, 2024 by wenscarl Loading…
Mark JAX git repository as safe in the asan workflow
#25115 opened Nov 26, 2024 by copybara-service bot Loading…
@nitins17
Add exec_time_optimization_effort and memory_fitting_effort flags.
#25114 opened Nov 26, 2024 by jedborovik Loading…
1
[AutoPGLE] Fix pgle test after removing pjit cache.
#25113 opened Nov 26, 2024 by copybara-service bot Loading…
Skip testAutoPgle tests.
#25112 opened Nov 26, 2024 by copybara-service bot Loading…
[Pallas][Mosaic GPU] Add support for compressing squeezed dims in async_copy + grid fixes
#25110 opened Nov 26, 2024 by copybara-service bot Loading…
@apaszke
[Mosaic TPU] Add support for modeling loads/stores and fix minor issues in model extraction
#25108 opened Nov 26, 2024 by copybara-service bot Loading…
@apaszke
Track mapping of platform aliases to compile-only backends
#25107 opened Nov 26, 2024 by jaro-sevcik • Draft
1
[Mosaic:TPU] Enable broadcast from 1-D vectors
#25106 opened Nov 26, 2024 by copybara-service bot Loading…
[Mosaic:TPU] Add relayout for adding minor implicit dim and relax some offset restrictions on similar shape cast
#25105 opened Nov 26, 2024 by copybara-service bot Loading…
Add lax.composite primitive
#25104 opened Nov 25, 2024 by copybara-service bot Loading…
[JAX] Add Python binding for building a colocated Python program
#25102 opened Nov 25, 2024 by copybara-service bot Loading…
[Pallas TPU] [WIP] Add vector support to pl.debug_print
#25099 opened Nov 25, 2024 by copybara-service bot Loading…
@ayaka14732
Start a new TPU interpret mode for Pallas
#25097 opened Nov 25, 2024 by jburnim Loading…
Update ROCm scripts to match new build.py usage pull ready Ready for copybara import and testing
#25096 opened Nov 25, 2024 by nitins17 Loading…
1
Reverts e72b449089f6af4ceb18288e36215b3c76e69245
#25092 opened Nov 25, 2024 by copybara-service bot Loading…
[cuda] Bump nvidia-cuda-nvcc-cu12 dependency to 12.6.85
#25091 opened Nov 25, 2024 by gspschmid Loading…
Bump hypothesis from 6.102.4 to 6.119.4 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25089 opened Nov 25, 2024 by dependabot bot Loading…
Update the "CI" workflow to use the new self-hosted runners
#25087 opened Nov 25, 2024 by copybara-service bot Loading…
@nitins17
3
[mgpu] FragentedArray.foreach() can now optionally return a new array
#25086 opened Nov 25, 2024 by copybara-service bot Loading…
Document cudaMallocAsync
#25082 opened Nov 25, 2024 by nouiz Loading…
[pallas_mgpu] Map indices primitive
#25078 opened Nov 25, 2024 by copybara-service bot Loading…
Test lax_scipy_spectral_dac error message
#25073 opened Nov 23, 2024 by copybara-service bot Loading…
Move tsl/platform/{build_config,build_config_root,rules_cc}.bzl to xla/tsl/platform
#25072 opened Nov 23, 2024 by copybara-service bot Loading…
[Mosaic TPU] Support packed type matmul with arbitrary shapes.
#25068 opened Nov 22, 2024 by copybara-service bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.