Skip to content
This repository was archived by the owner on Oct 31, 2025. It is now read-only.

Pull requests: google/trax

18 Open 1,563 Closed
18 Open 1,563 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 reviewsReview requiredApproved reviewChanges requested
Assignee
Filter by who’s assigned
Assigned to nobodyLoading
Sort
Sort by
NewestOldestMost commentedLeast commentedRecently updatedLeast recently updatedBest match
Most reactions
👍👎😄🎉😕❤️🚀👀

Pull requests list

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