Skip to content

Pull requests: google/trax

New pull request New
Clear current search query, filters, and sorts
18 Open 1,560 Closed
18 Open 1,560 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

Update decoding.py cla: yes
#1713 opened Oct 27, 2021 by sunvod
Add Pad layer with some Args cla: yes
#1715 opened Nov 5, 2021 by sunvod
ReformerLM api changes in text-gen notebook
#1723 opened Dec 9, 2021 by d0rc
Update trax.supervised.rst - This will now display documentation for all the classes defined inside the file
#1726 opened Jan 5, 2022 by vsnupoudel
1
tighten test_monte_carlo_bias_is_zero bound
#1727 opened Jan 25, 2022 by austereantelope
Fix issues related to new behavior of JAX DeviceArray.copy()
#1734 opened Apr 1, 2022 by copybara-service bot
Update inputs.py
#1761 opened Sep 26, 2022 by chrisrobertsoncypress
1
Disables a failing remat test.
#1763 opened Nov 4, 2022 by copybara-service bot
Move the TF-NumPy einsum test from Trax to the TF-NumPy directory.
#1777 opened May 26, 2023 by copybara-service bot
Nick dory patch 1
#1781 opened Aug 9, 2023 by NickDory
1
fix: write file typo
#1782 opened Aug 19, 2023 by 0o001
Update core.py: fix docstring "y = Wx + b" -> "y = xW + b"
#1784 opened Aug 29, 2023 by arvyzukai
1
Refactoring to 1.5.1
#1792 opened Nov 27, 2023 by mmarcinmichal
1
Replace references to deprecated device_buffer attributes
#1793 opened Dec 6, 2023 by copybara-service bot
Remove uses of deprecated jax.lax.tie_in
#1795 opened Jan 29, 2024 by copybara-service bot
Update references to JAX's GitHub repo
#1805 opened Dec 5, 2024 by copybara-service bot
Future-proof calls to jnp.solve on batched 1D inputs.
#1806 opened Jan 7, 2025 by copybara-service bot
Avoid use of deprecated xla_bridge.get_backend().live_buffers()
#1809 opened Feb 4, 2025 by copybara-service bot
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.