-
This is a great library first of all, so kudos to the developers !
My questions is: In the Package description in section `predict.gradient_descent_mse_ensemble` and you mention a paper as [11] but …
-
First, thank you for this handy library.
I have a question about ``stax.Aggregate`` function.
I tried to use ``neural_tangents.predict`` library with graph neural tangent kernel (i.e.``stax.aggrega…
-
Union types express a subtyping relationship, but I am unclear as to whether typeclasses (i.e. Rust's traits) do?
If a `trait B` extends another `trait A` and `B` reuses the implementations of `A`, c…
-
Please make sure that this is a feature request. As per our [GitHub Policy](https://github.com/tensorflow/tensorflow/blob/master/ISSUES.md), we only address code/doc bugs, performance issues, feature …
-
**System information**
- Have I written custom code (as opposed to using a stock example script provided in TensorFlow): yes
- OS Platform and Distribution (e.g., Linux Ubuntu 16.04): macOS 10.14.6
…
-
Provide as much information as possible. At least, this should include a description of your issue and steps to reproduce the problem. If possible also provide a summary of what steps or workarounds y…
-
Hi, I'm new to neural tangents. I don't quite understand the output dimensions in the examples.
As my understanding, the kernel size calculated by `kernel_fn` should be `N*N`, and the example for …
-
### 🐛 Describe the bug
I am trying to create a new class `LinearizedModel` that takes as input an `nn.Module` and creates a linearized version of it that can be trained transparently as any other `…
-
Hey guys!
I hope you are doing great. I faced an error while working with jax.linalg which I don't understand how to fix or which part it's related to. I mentioned the error and the details in the ot…
-
Hi
I am new to JAX, and my question is can we use this package to train regular neural networks without the NTK notion? If this is the case, are there any examples or a document to start with?
T…