issues
search
lhnguyen102
/
cuTAGI
CUDA implementation of Tractable Approximate Gaussian Inference
MIT License
24
stars
9
forks
source link
Probabilistic activations
#23
Closed
lhnguyen102
closed
1 year ago
lhnguyen102
commented
1 year ago
Add probabilistic activation functions (CUDA & CPU)
Mixture relu, sigmoid, and tanh
Localized linear softmax and remax
Improve stablility for a larger batch size
Add C++ Unittest for fnn and full covariance