zacmar / logsumexpv2 Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

efficient CUDA implementations of the LogSumExp activation function and sampling routines for categorical and generalized inverse Gaussian distributions.

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 34 Commits
logsumexpv2		logsumexpv2
README.md		README.md
act.cpp		act.cpp
act_kernel.cu		act_kernel.cu
bench_lse.py		bench_lse.py
impl.py		impl.py
setup.py		setup.py
test-gig.py		test-gig.py
test-multisigma.py		test-multisigma.py
test_interface.py		test_interface.py
test_logsumexp.py		test_logsumexp.py

Repository files navigation

LogSumExp activation function

LogSumExp activation function

Installation

pip install .

About

efficient CUDA implementations of the LogSumExp activation function and sampling routines for categorical and generalized inverse Gaussian distributions.

Report repository

Releases

No releases published

Packages

No packages published

Contributors 2

Languages