trax

trax copied to clipboard

Fixes #1744 I'd contributed the original notebook with a link to my branch of the repository rather than it's expected path. CC: @afrozenator

SauravMaheshkar

Trax.data API explained -> Open in Colab errors

1

### Description In the README, following the first link to [trax.data API explained](https://github.com/google/trax/blob/master/trax/examples/trax_data_Explained.ipynb) : When clicking on "Open in Colab" produces: 

ras44

ValueError: no signature found for builtin function

4

### Description I'm using Tensorflow on my M1 Mac, but importing Trax throws this error. I've tried envs with Tensorflow versions 2.5.0, 2.6.0, and 2.7.0. ### Environment information ``` OS:...

alistairmayo

### Description I got this error while importing the trax $ pip freeze | grep trax ``` trax 1.3.1 ``` $ pip freeze | grep tensor ``` mesh-tensorflow==0.1.21 tensor2tensor==1.15.7 tensorboard==2.9.1...

NirajanBekoju

Not able to import trax in google colab though trax is installed

1

### Description I am trying trax-ml library in Google Colab. I installed trax using below command !pip install trax==1.3.1 But I am not able to import trax library. Please help...

ankusri

Regarding Reformer: [paper](https://arxiv.org/pdf/2001.04451.pdf) | [code](https://github.com/google/trax/tree/master/trax/models/reformer) From paper: > .. show that it performs the same as the normal Transformer when using the same number of parameters; we achieve this by...

alexm-gc

How to implement NER model using trax

5

How to use Trax to implement NER model in NLP using reformer.

omerarshad

Trax ML: GPU memory allocated but completed on CPU

5

### Description Hi, I have run the example ende translation script and have installed jax+cuda so that I do *not* get the typical "No GPU/TPU found, falling back to CPU"...

ghost

Fix issues related to new behavior of JAX DeviceArray.copy() In https://github.com/google/jax/pull/10069, JAX changes the behavior of DeviceArray.copy() so that it returns a DeviceArray rather than returning a numpy array. For...

copybara-service[bot]

Trax Throw error while installing on MAC book pro with M1 chip

9

### Description I created and environment using miniconda supported with M1 chip, after that we installed Tensorflow2.4 which can use GPU and tried installing Trax but it throws following error....

pratik-pk