Is the module trax.fastmath of TRAX package deprecated? I am using this module, but ModuleNotFoundError is returned.
I replaced trax.fastmath with trax.math and things worked perfectly fine. I am still not sure if this has been updated in the Trax documentation.
I replaced trax.fastmath with trax.math and things worked perfectly fine. I am still not sure if this has been updated in the Trax documentation.