Dmitry Kostyaev
Results
2
comments of
Dmitry Kostyaev
To prevent preallocation of gpu memory triggered by `jnp.ones([1])` I had to escape quotes in the env variable value to make it work: `os.environ["XLA_PYTHON_CLIENT_ALLOCATOR"] = '\"platform\"' ` or alternatively in...
@lhegen tensorflow implementation is here: https://github.com/kostyaev/ICNR