google neural-tangents issues

google / neural-tangents

Fast and Easy Infinite Neural Networks in Python

https://iclr.cc/virtual_2020/poster_SklD9yrFPS.html

Apache License 2.0

2.28k stars 226 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

Error with batching

#206 HaozhenZhao closed 2 months ago
0
Support python 3.12 and misc

#205 romanngg opened 2 months ago
1
High Memory Usage on Infinite-Width NTK for GPU Only

#204 deoliveirajoshua opened 3 months ago
1
Transformer Block Implementation gives NotImplementedError

#203 esnvidia opened 4 months ago
0
Tag a release

#202 PhilipVinc opened 4 months ago
2
Calling the empirical kernel function with different parameters returns same result

#201 PhilippMisofCH closed 7 months ago
0
Question regarding the cookbook

#200 XinyiYS opened 9 months ago
0
Inefficient jacobian computation for embedding layers.

#199 mohamad-amin opened 9 months ago
1
NKT_mean output Nan, when the number of training sample is increased

#198 zhangbububu opened 10 months ago
3
NTK/NNGP behavior in the infinite regime when weights are drawn from Gaussians with high standard deviation

#197 tengandreaxu opened 11 months ago
7
How to use batch to gradient_descent_mse_ensemble ?

#196 yanglebupt closed 6 months ago
1
stax.serial PSDness

#195 wang-fang-cv closed 9 months ago
1
NTK is not PD

#193 bayesfourrier closed 12 months ago
0
using stax.Cos(a=1.0, b=1.0, c=0.0) to get kernel from conv layer gives error

#192 bayesfourrier closed 12 months ago
2
Erf function goes beyond [-1,1]

#191 bangxiangyong opened 1 year ago
2
pip install issues

#190 SamTov closed 1 year ago
2
How to compute the empirical after kernel?

#189 VMS-6511 opened 1 year ago
1
How to obtain aleatoric uncertainty?

#188 bangxiangyong opened 1 year ago
2
Colab notebooks issue

#187 juan190199 closed 1 year ago
2
eNTK implementation uses deprecated xla attribute

#186 shawn-im closed 1 year ago
2
Correct arxiv links in README.md

#185 mmhamdy closed 1 year ago
3
Question regarding lr in Neural Tangents Cookbook

#184 alicedimo closed 1 year ago
0
Question regarding OOM issues

#182 MichaelMMeskhi opened 1 year ago
3
Question: Connection MLE "parametrized" GP in infinite Width Limit vs minimizing MSE "parametrized" Kernel in infinite Width

#181 yCobanoglu closed 1 year ago
4
Question: Relu Kernel Computation

#180 yCobanoglu closed 1 year ago
3
The analytical output of GP can not fit the result of NNGP generated by the nt.predict.gp_inference

#178 Gengfu-He opened 1 year ago
1
How to do Aggregate on a Graph whose nodes are all vectors

#176 SchenbergZY opened 1 year ago
6
Report of Abuse of misinformation in scientific research

#175 yhyu13 closed 1 year ago
0
Are the parameters of the elementwise nonlinear layers trainable?

#174 rishiedupalli closed 1 year ago
4
AttributeError: module 'neural_tangents' has no attribute 'utils'

#173 LeavesLei opened 1 year ago
8
Defining Own Kernel Function in Neural Tangents

#172 snehjp2 opened 1 year ago
1
Conflicting dependencies between Jax and neural-tangents

#171 snehjp2 closed 1 year ago
6
The different output dimensions in examples

#170 cycyyy closed 1 year ago
2
Cases where NTK and NNGP are very different

#169 YehonatanAvidan opened 1 year ago
0
Excessive memory consumption for deep networks

#168 jglaser opened 2 years ago
2
Questions about memory consumption of infinitely wide NTK

#166 jasonli0707 opened 2 years ago
6
Does neural-tangents work for custom layer?

#165 Shuhul24 opened 2 years ago
1
Draw Phase Diagram for CNTK

#164 hhorace opened 2 years ago
1
Question: simple example poor performance, what am I doing wrong?

#163 mfouesneau opened 2 years ago
5
neural tangents vs flax

#162 po-oya closed 2 years ago
3
Why is the covariance a 2D matrix?

#160 LeonhardStorm opened 2 years ago
2
Feature masks do not get reduced in the kernel

#159 jglaser opened 2 years ago
2
Fix masking with more than one input feature

#158 jglaser opened 2 years ago
1
Use local device count for pmap in np.batch

#157 jglaser closed 2 years ago
2
Point JAX installs in colabs to the new CUDA releases page.

#156 romanngg closed 2 years ago
0
Neural tangents finite network gives different results compared with JAX

#155 NiuJ1ao closed 2 years ago
1
Improve type annotations and documentation

#154 romanngg closed 2 years ago
1
predict.py raises an error

#152 zeinabsadeghian closed 1 year ago
7
How to analyze transductive node classification

#151 joneswong opened 2 years ago
2
Question about training model

#150 Firegreat123 closed 2 years ago
2