•  


Pull requests · google/jax · GitHub
Skip to content

Pull requests: google/jax

New pull request New
328 Open 13,337 Closed
328 Open 13,337 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ? + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Make attrs work with pytrees pull ready Ready for copybara import and testing
#21478 opened May 29, 2024 by dougalm Loading…
1
Improve docs for jnp.partition & argpartition documentation
#21476 opened May 29, 2024 by jakevdp Loading…
@jakevdp
Add a link to another page with XLA flags.
#21473 opened May 29, 2024 by nouiz Loading…
For MATT: x64 in jaxpr eqn context
#21472 opened May 28, 2024 by copybara-service bot Loading…
@yashk2810
[array api] update test suite to most recent commit
#21469 opened May 28, 2024 by jakevdp Loading…
@jakevdp
jnp.searchsorted: support sorter argument
#21468 opened May 28, 2024 by jakevdp Loading…
@jakevdp
[Mosaic] Fix bug in VectorLayout::generalizes introduced in cl/636250759
#21464 opened May 28, 2024 by copybara-service bot Loading…
Makes global_shape optional for jax.make_array_from_process_local_data.
#21456 opened May 28, 2024 by copybara-service bot Loading…
[Mosaic GPU] Add a synchronization point at the end of the constructor for BarrierArray .
#21450 opened May 28, 2024 by copybara-service bot Loading…
@bchetioui
Add the function name, the Jaxpr, and lowering platforms to stages.Lowered pull ready Ready for copybara import and testing
#21447 opened May 28, 2024 by gnecula Loading…
@gnecula
20
Bump zipp from 3.18.1 to 3.19.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#21445 opened May 27, 2024 by dependabot bot Loading…
Added trace alias to jnp.linalg
#21442 opened May 27, 2024 by vfdev-5 • Draft
@jakevdp
4
Updated jnp.ceil/floor/trunc to preserve int dtypes
#21441 opened May 27, 2024 by vfdev-5 • Draft
1
Added op name and dtype into testcase name for tests in lax_numpy_operators_test.py
#21440 opened May 27, 2024 by vfdev-5 • Draft
Add reference to Flax NNX
#21439 opened May 27, 2024 by cgarciae Loading…
Support Hessian of gamma-distributed samples
#21432 opened May 25, 2024 by NeilGirdhar Loading…
1
remove is_training from cuDNN SDPA API pull ready Ready for copybara import and testing
#21430 opened May 24, 2024 by Cjkkkk Loading…
4
[Mosaic GPU] Add the option to control output dtype in the matmul examples
#21427 opened May 24, 2024 by copybara-service bot Loading…
[Mosaic GPU] Support for different accumulator types
#21426 opened May 24, 2024 by copybara-service bot Loading…
[Mosaic GPU] Integrate the ExecutionEngine with the jaxlib GPU plugin
#21425 opened May 24, 2024 by copybara-service bot Loading…
@apaszke
[XLA][Mosaic] Add support for fp8 matmuls
#21419 opened May 24, 2024 by copybara-service bot Loading…
DOC: Add code examples to jax.scipy.fft.dct doc
#21418 opened May 24, 2024 by rajasekharporeddy Loading…
@jakevdp
[Mosaic] Packed loads and stores with 1D tiling should use (1, 128 * packing)
#21417 opened May 24, 2024 by copybara-service bot Loading…
print literal types in jaxprs pull ready Ready for copybara import and testing
#21415 opened May 24, 2024 by mattjj Loading…
Clear caches of sharding._addressable_devices_indices_map in api.clear_backends and let api.clear_backends be called before process terminates. This could make all living C++ PjRt client objects to be successfully destroyed during Python garbage collection.
#21413 opened May 24, 2024 by copybara-service bot Loading…
@yueshengys
ProTip! Type g i on any issue or pull request to go back to the issue listing page.
- "漢字路" 한글한자자동변환 서비스는 교육부 고전문헌국역지원사업의 지원으로 구축되었습니다.
- "漢字路" 한글한자자동변환 서비스는 전통문화연구회가 "울산대학교한국어처리연구실 옥철영(IT융합전공)교수팀"에서 개발한 한글한자자동변환기를 바탕하여 지속적으로 공동 연구 개발하고 있는 서비스입니다.
- 현재 고유명사(인명, 지명등)을 비롯한 여러 변환오류가 있으며 이를 해결하고자 많은 연구 개발을 진행하고자 하고 있습니다. 이를 인지하시고 다른 곳에서 인용시 한자 변환 결과를 한번 더 검토하시고 사용해 주시기 바랍니다.
- 변환오류 및 건의,문의사항은 juntong@juntong.or.kr로 메일로 보내주시면 감사하겠습니다. .
Copyright ⓒ 2020 By '전통문화연구회(傳統文化硏究會)' All Rights reserved.
 한국   대만   중국   일본 

Close