trax

trax copied to clipboard

Nick dory patch 1

1

NickDory

### Description I'm starting to test Trax. Is possible Linformer algorithm in future? In Linformer paper I read that it is faster than Reformer. ### Environment information OS: Linux Mint...

andr1972

Hi, I am facing an issue and I was hoping someone could tell me how to solve it. I am trying to use a neural network sentiment analysis like it...

SisypheGeek

### Description Hi, I would like to install trax locally. Firstly, I found jax I installed is not suitable for GPU, so I follow the jax github to install Cuda...

LiuZhenshun

Move the TF-NumPy einsum test from Trax to the TF-NumPy directory.

copybara-service[bot]

TypeError: unsupported operand type(s) for ==: 'Array' and 'tuple'

2

### Description ... ### Environment information ``` OS: $ pip freeze | grep trax trax==1.4.1 $ pip freeze | grep tensor tensorboard==2.11.2 tensorboard-data-server==0.6.1 tensorboard-plugin-wit==1.8.1 tensorflow==2.11.0 tensorflow-datasets==4.8.2 tensorflow-estimator==2.11.0 tensorflow-hub==0.12.0 tensorflow-io-gcs-filesystem==0.30.0 tensorflow-metadata==1.12.0...

mbateman

AttributeError: module 'jax.ops' has no attribute 'index_add'

1

### Description I am trying to do something basic in my code: ```python import numpy as np # regular ol' numpy from trax import layers as tl # core building...

cmosguy

Unable to import trax

1

### Description I installed Trax but cannot import ### Environment information ``` OS: Ubuntu 20.04.5 LTS $ pip freeze | grep trax trax==1.4.1 $ pip freeze | grep tensor tensorboard==2.11.2...

JoseJamesMD

### Description When I have set a TF backend (2.11.0) to computation and try to use SelfAttention from research module than I receive Exception below: File "/usr/local/lib/python3.10/dist-packages/trax/layers/research/efficient_attention.py", line 1536, in...

mmarcinmichal

### Description How to limit the datasets from TFDS when I get from 'opus/medical' ### Environment information ``` OS: Google collab

oendnsk675