issues
search
google
/
trax
Trax — Deep Learning with Clear Code and Speed
Apache License 2.0
8.07k
stars
813
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
[JAX] Replace users of deprecated jax.lib.xla_bridge functions with more public names.
#1704
copybara-service[bot]
closed
2 years ago
0
Update NMT_with_Transformers_Reformers_using_Trax.ipynb
#1703
sunvod
closed
2 years ago
0
Update NMT_with_Transformers_Reformers_using_Trax.ipynb
#1702
sunvod
closed
2 years ago
2
[JAX] Replace uses of deprecated `jax.ops.index_update(x, idx, y)` APIs with their up-to-date, more succinct equivalent `x.at[idx].set(y)`.
#1701
copybara-service[bot]
closed
3 years ago
0
Add depthwise convolution and allow to use it in CausalAttention.
#1700
lukaszkaiser
closed
2 years ago
0
Hourglass - evaluation colabs for enwik8 and downsampled imagenet
#1699
syzymon
closed
2 years ago
0
Add HourglassLM
#1698
syzymon
closed
2 years ago
0
RAM crash while optimizing a linear regression model
#1697
aycandv
opened
3 years ago
0
Make T5 imports dynamic so it can be optional.
#1696
lukaszkaiser
closed
3 years ago
0
Relu layer nested in its own Serial combinator by default
#1695
kbvw
opened
3 years ago
1
Wiki40b gin config update
#1694
syzymon
closed
3 years ago
0
Ensure values passed to jax.numpy functions are arrays rather than lists.
#1693
copybara-service[bot]
closed
3 years ago
0
[TRAX] Also install t5 while running tests.
#1692
copybara-service[bot]
closed
3 years ago
0
Make installation of TRAX easier.
#1691
copybara-service[bot]
closed
3 years ago
0
Update Relformer enwik8 gin config
#1690
syzymon
closed
3 years ago
0
A fix to a regression test.
#1689
copybara-service[bot]
closed
3 years ago
0
Remove 'xla' as the default for jax_xla_backend flag.
#1688
copybara-service[bot]
closed
3 years ago
0
Rotary embeddings
#1687
PiotrNawrot
closed
3 years ago
3
Multiple heads option is not working in SelfAttention
#1686
kenenbek
opened
3 years ago
1
Reformer does not export to SavedModel format.
#1685
StewartSethA
opened
3 years ago
0
Rotary embeddings
#1684
PiotrNawrot
closed
3 years ago
3
RelformerLM - final implementation + gin configs
#1683
syzymon
closed
3 years ago
6
How to make predictions on a trained model
#1682
TurquoiseA
opened
3 years ago
0
Wrapping up a combinator layer
#1681
aycandv
closed
3 years ago
2
[JAX] Fix test failure in Trax due to upcoming change to JAX.
#1680
copybara-service[bot]
closed
3 years ago
0
add encoder decoder model
#1679
MaanasArora
closed
2 years ago
4
Nicer printing in Terraformer predict script.
#1678
lukaszkaiser
closed
3 years ago
0
Scientific papers decoding in eval mode.
#1677
lukaszkaiser
closed
3 years ago
0
Add eval mode to decoding, no-positional encoding option.
#1676
lukaszkaiser
closed
3 years ago
2
Trax ML: GPU memory allocated but completed on CPU
#1675
ghost
opened
3 years ago
5
making the Terraformer predict script compatible with scipap dimensions.
#1674
copybara-service[bot]
closed
3 years ago
3
Fix Tensordot test by adjusting tol.
#1673
copybara-service[bot]
closed
3 years ago
0
Fix tf.experimental.numpy.linspace step output, now it's consistent with Numpy.
#1672
copybara-service[bot]
closed
3 years ago
0
Fix testTensordot test by adjusting tol.
#1671
copybara-service[bot]
closed
3 years ago
0
ValueError on predict mode Transformer model
#1670
shashank2123
opened
3 years ago
2
L2 & L1 Regularizers in Trax
#1669
ghost
opened
3 years ago
0
Added a Python script running predict mode of pretrained Terraformer model.
#1668
copybara-service[bot]
closed
3 years ago
0
MathQA-Python generation colab.
#1667
copybara-service[bot]
closed
3 years ago
0
Replacing scipy.misc with scipy.special in MathQA processing.
#1666
copybara-service[bot]
closed
3 years ago
0
additional infrastructure for the MathQA-Python dataset.
#1665
copybara-service[bot]
closed
3 years ago
0
LSHSelfAttention only works with different lengths if the parameter use_reference_code is set to "True"
#1664
renevs
opened
3 years ago
0
A colab for evaluation of sparse models.
#1663
copybara-service[bot]
closed
3 years ago
0
Test and challenge datasets for MathQA.
#1662
copybara-service[bot]
closed
3 years ago
0
Trax Import Error
#1661
memora0101
opened
3 years ago
0
tl.Relu and tl.ShiftRight layers gets inside Serial Combinator
#1660
apoorvgupta25
closed
3 years ago
1
Allow to configure checkpoint_low_metric and checkpoint_high_metric for training.Loop
#1659
manifest
closed
3 years ago
0
Add trax.data.Dup combinator
#1658
manifest
closed
3 years ago
0
Fix predict mode in PositionalEncoding with d_feature.
#1657
copybara-service[bot]
closed
3 years ago
0
Allow PretrainedBERT to set attributes required for loading the model from its checkpoints
#1656
manifest
closed
2 years ago
0
BERT model fails on its initialization
#1655
manifest
opened
3 years ago
2
Previous
Next