-
HI, Thank you for the nice work. I am trying to do kernel ridge regression with ntk. to do that i need down and upsampling layers. the following code i am using.
https://gist.github.com/jaya-pret/…
-
**What I need help with / What I was wondering**
I am unable to load a tensorflow dataset in my local environment given the following error:
_RecursionError: maximum recursion depth exceeded_
*…
-
Hi,
I am trying to generate conv kernel using the infinite network example given in the documentation .
https://neural-tangents.readthedocs.io/en/latest/stax.html
I am trying to add a activatio…
-
AutoEnzyme should probably be specialized and not fall back to DI.
In addition to being slower in some cases, it's been shown to cause errors (even segfaults) when using AutoEnzyme in DI whereas us…
-
Hi, I am running Jax v0.4.1 (for CPU) and neural-tangents v0.5.0. When importing neural-tangents in a script, I encounter this error:
`ImportError: cannot import name 'tree_multimap' from 'jax.tre…
-
It seems that the version of NT that is on PyPi is out of date as it throws Jax errors. Could you make a release? It isn't an issue for using NT but we have some software built on top of it and now al…
-
Dear All,
I believe there is some problem when importing neural-tangents. When I ran the code in colab, I got the message below.
Thank you,
Juan
```
----------------------------------------…
-
Hey guys!
I hope you're doing well:)
So I installed the neural tangent as the instructions in the https://pypi.org/project/neural-tangents/. But when I try to run the tests in the link, this error r…
-
I was using the eNTK implementation when I ran into this error
`module 'jax.interpreters.xla' has no attribute 'xla_call_p'`
and it seems that xla_call_p has been deprecated after jax 0.4.10. I beli…
-
I have [a long comment](https://users.rust-lang.org/t/rust-as-a-high-level-language/4644/72) at the Rust forum (some of the inspiration came from @keean linking me to Sean Parent's [video](https://you…