Example code to create a machine translation using trax transformer model.

Description

Hi, I am little new to this area. Please excuse me if I am asking any stupid question. I wanted to have machine translation model using Transformers and the model is trained on my own data. Can I get an example code where I can pass my own data as input (both input and target) to the trax transofrmers so that it can predict the translated value.

I will be using my own laptop for preparing and running the model. It does not have GPU.

Also if possible can I get an example code for speech recognition as well using the transformer model.

Thanks Nagaraju

…

Environment information

OS: <win10>

$ pip freeze | grep trax
# your output here

$ pip freeze | grep tensor
# your output here

$ pip freeze | grep jax
# your output here

$ python -V
# your output here

For bugs: reproduction and error logs

# Steps to reproduce:
...

# Error logs:
...

[lukaszkaiser]

It’s a very good idea - we’ll try to add a colab soon doing just that!

[nag0811]

@lukaszkaiser Thanks for the response 👍 Unfortunately I am not able to run the code on my windows laptop. I tried to install trax using pip install trax. But it failed at jaxlib with following error. ERROR: Could not find a version that satisfies the requirement tensorflow-text (from trax) (from versions: none) ERROR: No matching distribution found for tensorflow-text (from trax)

In the notebook, I could see trax is listed as module but when I run the code it is giving “No module named trax”