Skip to content

/ trax

New pull request New
14 Open 945 Closed
14 Open 945 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels.
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
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Fix data names typo
cla: yes
#1056 opened Sep 26, 2020 by weiddeng
2
Add multisync and gated Q to the QWR baseline config.
cla: yes
#1053 opened Sep 25, 2020 by trax-robot
Test memory usage of ReformerLM with FAVOR.
cla: yes
#1052 opened Sep 25, 2020 by trax-robot
[TF-numpy/extensions] Moves the single-output-as-list workaround for tpu.replicate to within tf.function, because accessing the global variable _orig_result_is_list outside tf.function (as we do now) is not thread-safe.
cla: yes
#1051 opened Sep 25, 2020 by trax-robot
Add code highlighting to README.md
cla: yes
#1043 opened Sep 23, 2020 by AmenRa
2
Add layer norm to PureMLP.
cla: yes
#1041 opened Sep 23, 2020 by trax-robot
Assert Shape interface
cla: yes
#1037 opened Sep 22, 2020 by trax-robot
Use jax.api.device_put_sharded() in place of private JAX apis.
cla: yes
#1017 opened Sep 16, 2020 by trax-robot
[TF-numpy] Adds XLA test to lax_numpy_test.
cla: yes
#1012 opened Sep 15, 2020 by trax-robot
Confidence-threshold and basic-controller driven layer dropping.
cla: yes
#1001 opened Sep 7, 2020 by trax-robot
a Layer adding metrics to TensorBoard
cla: yes
#987 opened Aug 31, 2020 by trax-robot
Move RL agents to a separate package.
cla: yes
#977 opened Aug 26, 2020 by trax-robot
Add a unit test for the Attention layer.
cla: yes
#854 opened Jul 20, 2020 by trax-robot
Added nonlinear attention
cla: yes
#769 opened Jun 25, 2020 by JacobFV
3
ProTip! Mix and match filters to narrow down what you’re looking for.
You can’t perform that action at this time.