-
Hello! I was wondering if it's possible for one to write their own `kernel_fn` in neural tangents to do regular kernel regression (for example, a Gaussian kernel). Naively just writing a `new_kernel_f…
-
I have built a custom layer (`KerasLayer`) using `class` in python (say `class NewLayer`). Can I use something like `stax.NewLayer` for manipulating `neural-tangents` on this custom layer?
-
Hi, nice work, and thanks for sharing the code. When I was running the code, we encountered the following error.
```shell
jax._src.traceback_util.UnfilteredStackTrace: AttributeError: 'ShapedArra…
-
I'm trying to differentiate through `predict_fn` provided by https://github.com/google/neural-tangents. This is doable with `jax.grad`, but not with `eagerpy.value_and_grad`.
ybj14 updated
4 years ago
-
Hi developers, I've met a problem when using neural-tangents as follows:
```
KERNEL_FN = nt.utils.batch.batch(KERNEL_FN, batch_size=kernel_batch_size)
AttributeError: module 'neural_tangents' has…
-
I am observing an error message when providing masked inputs with more than one feature dimensions to a kernel that involves `stax.GlobalAvgPool()`
Reproducer:
```
import jax
import jax.numpy as…
-
When I use neural-tangents 0.3.2 to run the 1d_regression.ipynb, I got ValuError:
Co…
-
I'm curious about the initialization for CNTK, so I replace the `kernel_fn` in `c_map(W_var, b_var)` function in [colab](https://colab.research.google.com/github/google/neural-tangents/blob/main/noteb…
-
Hey!
I was reading through the code and I noticed that you're using element-wise exponential matrix here:
https://github.com/google/neural-tangents/blob/5f286b7696364217aa4a2d92378aabd0203a791e/n…
-
Hello,
I implemented a brutally simple infinite-width model, calling the kernel_fn with a batch of a single vector.
When I run this on CPU, I don't run into any exorbitant memory issues.
How…