Skip to content

Pull requests: google/trax

New pull request New
15 Open 1,556 Closed
15 Open 1,556 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

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