trax

trax copied to clipboard

Reformer Model for Speech Recognition

5

Coming from tensor2tensor I was wondering whether the Reformer model would be also a candidate for speech recognition? Looking at the examples there is none for ASR. Would it be...

stefan-falk

### Not an issue but a request Hi, While using Trax for Text summarization, I downloaded and edited [text_generation.ipynb](https://github.com/google/trax/blob/master/trax/models/reformer/text_generation.ipynb) notebook by feeding inputs of the form [doc][summary] and loss mask...

ZohebAbai

### Description train_rl fails on different issues, depending on tensorflow version. The version of Python (tested 3.7 and 3.8) does not seem to play a role. A Traceback for each...

davidjbuerger

Maintained Documentation

6

### Description Trax is a library for deep learning that focuses on sequence models and reinforcement learning. It combines performance with code clarity and maintained documentation and tests. ... Sorry...

felipeboffnunes

Pre-trained Reformer models

1

It would be great if pretrained Reformer models become available (e.g., trained on BooksCorpus and English Wikipedia).

markusdr

### Description I think in [`resnet.py`](https://github.com/google/trax/blob/0d1fa5db040e50659f5f10da92452183481068ae/trax/models/resnet.py#L109), the padding option for `MaxPool` should be `'SAME'`. The shape of the output of `MaxPool` and the `Resnet50ConvBlock` right after it becomes `B x...

ryotatomioka

### Description Hi, I am trying to follow this tutorial: https://github.com/google/trax/blob/master/trax/examples/NMT_with_Transformers_Reformers_using_Trax.ipynb Setting the runtime to TPU on Colab used to work couple of days ago. But now it crashes with...

agoliaei

### Description Small issue here: The Open in Colab button in the notebook Knowledge_Tracing_Transformer.ipynb in the directory "trax/trax/examples" leads to a private file on Google Drive.

haytamdon

Cannot import Trax

8

### Description I am trying to use trax following NLP specialization but couldn't import it. I have search for solution and ask help on ChatGPT but could not solve it....

Nuna7

Remove uses of deprecated jax.lax.tie_in This has been a no-op since jax v0.2.0, and passes the second argument through unchanged. `tie_in` will be deprecated as of jax v0.4.24; see https://github.com/google/jax/pull/19413

copybara-service[bot]