This repository was archived by the owner on Oct 31, 2025. It is now read-only.
- Notifications
You must be signed in to change notification settings - Fork 828
Pull requests: google/trax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Avoid use of deprecated xla_bridge.get_backend().live_buffers()
#1809 opened Feb 4, 2025 by copybara-service bot Loading…
Future-proof calls to jnp.solve on batched 1D inputs.
#1806 opened Jan 7, 2025 by copybara-service bot Loading…
Replace references to deprecated device_buffer attributes
#1793 opened Dec 6, 2023 by copybara-service bot Loading…
Update core.py: fix docstring "y = Wx + b" -> "y = xW + b"
#1784 opened Aug 29, 2023 by arvyzukai Loading…
Move the TF-NumPy einsum test from Trax to the TF-NumPy directory.
#1777 opened May 26, 2023 by copybara-service bot Loading…
Fix issues related to new behavior of JAX DeviceArray.copy()
#1734 opened Apr 1, 2022 by copybara-service bot 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…
ProTip! Mix and match filters to narrow down what you’re looking for.