I'm pretty new to working with Trax, a framework built by the Google Brain team to work with deep learning models as an alternative to TensorFlow. As ...

How can I code my own MaxPooling_1D layer in google Trax? I understand that current max pooling is based on 2D max pooling. Here's how I tried using ...

I was stumped the first time I loaded this library. In my local computer it tooks me at least 40s to load trax on a local Jupyter Notebook and more th ...

trax New to trax, I'm trying to run it locally (macOS 12.1, Apple Silicon ARM M1 processor, 8GB RAM, Anaconda), but I'm running into some issues. In ...

I am trying to build an attention model but Relu and ShiftRight layer by default nested inside the Serial Combinator. This further gives me errors in ...

I have trained an off-the-shelf Transformer(). Now I want to use the encoder in order to build a classifier. For that I want to only use the first to ...

Is the module trax.fastmath of TRAX package deprecated? I am using this module, but ModuleNotFoundError is returned. ...

I am working on a project involving Google's TRAX package. I encountered the following error message while running a snippet of code on Jupyter notebo ...

Anyone managed to log the loss with TensorBoard? I am using the trax ml library. I am getting this error TypeError: 'SummaryWriter' object is not call ...

I would like to create a custom tensorflow dataset for summarization task. I have a set of reports with three gold summaries for every report. All the ...

I have been trying to create a vocab file in a nlp task to use in tokenize method of trax to tokenize the word but i can't find which module/library t ...

(Trax newbie here) I have a Trax loop object, from which I would like to extract a task object. The code below returns the error AttributeError: 'Loop ...

The AttentionQKV layer implemented by Trax is as the following: AttentionQKV In particular, what is the purpose of the three parallel dense layers? ...

How do I set up a multi-variate regression problem using Trax? I get AssertionError: Invalid shape (16, 2); expected (16,). from the code below, comi ...

I am new to Machine Translation and Trax. My goal is to understand the introductory example on transformers in Trax, which can be found at https://tra ...

I have been using TensorRT and TensorFlow-TRT to accelerate the inference of my DL algorithms. Then I have heard of: JAX https://github.com/googl ...