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

OS: <your answer here>

$ 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

# Steps to reproduce:
...!pip install trax==1.3.1
import trax

# Error logs:
...
```---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
[<ipython-input-3-3afdcd9c7c68>](https://localhost:8080/#) in <module>()
----> 1 import trax

2 frames
[/usr/local/lib/python3.7/dist-packages/trax/layers/__init__.py](https://localhost:8080/#) in layer_configure(*args, **kwargs)
     41 def layer_configure(*args, **kwargs):
     42   kwargs['module'] = 'trax.layers'
---> 43   return gin.external_configurable(*args, **kwargs)
     44 
     45 # pylint: disable=used-before-assignment

TypeError: external_configurable() got an unexpected keyword argument 'blacklist'