Index

_

_wait() (waitable.Waitable method)

A

args (context_switch.ParticleCache attribute) (context_switch.ParticleCacheLRU attribute) (messages.ReceiveFuncMSG attribute) (messages.ReceiveFuncPDMSG attribute)

B

bayes_infer() (push.bayes.ensemble.Ensemble method) (push.bayes.infer.Infer method) (push.bayes.stein_vgd.SteinVGD method) (push.bayes.swag.MultiSWAG method)

C

cache_size (context_switch.ParticleCache attribute) (context_switch.ParticleCacheLRU attribute) contains() (context_switch.ParticleCache method) (context_switch.ParticleCacheLRU method) context_switch module

create() (context_switch.ParticleCache method) (context_switch.ParticleCacheLRU method) create_optimizer() (in module push.bayes.ensemble) (in module push.bayes.swag)

D

detach_to_cpu() (in module utils) detach_to_device() (in module utils)

device (context_switch.ParticleCache attribute) (context_switch.ParticleCacheLRU attribute) (messages.ReceiveParticleInitPDMSG attribute)

E

Ensemble (class in push.bayes.ensemble)

F

fid (messages.ReceiveGetAckMSG attribute) flatten() (in module push.bayes.utils)

fn (messages.ReceiveRegisterPDMSG attribute) forward() (node_event_loop.NodeEventLoop method) (push.particle.Particle method)

G

get() (node_event_loop.NodeEventLoop method) (push.particle.Particle method)

get_var() (push.bayes.infer.Infer method)

I

in_queues (messages.NELBroadcastParticlesMSG attribute)

Infer (class in push.bayes.infer) init_node_event_loop() (in module push.push)

L

lock (context_switch.ParticleCache attribute)

M

messages module mk_empty_optim() (in module push.bayes.stein_vgd) mk_empty_scheduler() (in module push.bayes.ensemble) (in module push.bayes.stein_vgd) mk_module (context_switch.ParticleCache attribute) (context_switch.ParticleCacheLRU attribute) mk_optim (messages.ReceiveParticleInitPDMSG attribute) mk_optim() (in module push.bayes.ensemble) (in module push.bayes.swag) mk_scheduler() (in module push.bayes.ensemble) (in module push.bayes.swag) module context_switch messages node_event_loop push.bayes.ensemble push.bayes.infer push.bayes.stein_vgd push.bayes.swag push.bayes.utils push.particle push.pfuture push.push utils waitable

MSG (class in messages) msg (messages.ReceiveFuncPDMSG attribute) (messages.ReceiveRegisterPDMSG attribute) msg_name (messages.ReceiveFuncMSG attribute) MultiSWAG (class in push.bayes.swag)

N

NELBroadcastParticlesAckMSG (class in messages) NELBroadcastParticlesMSG (class in messages) NELSaveModel (class in messages) NELSaveModelAckPDMSG (class in messages) node_event_loop module

NodeEventLoop (class in node_event_loop) NodeEvtLoopCleanupMSG (class in messages) NodeEvtLoopInitMSG (class in messages) None (waitable.Waitable attribute) normal_prior() (in module push.bayes.stein_vgd) num_particles() (push.push.PusH method)

O

other_particles() (push.particle.Particle method)

out_queues (messages.NELBroadcastParticlesMSG attribute)

P

p_create() (push.push.PusH method) p_launch() (push.push.PusH method) p_parameters() (push.bayes.infer.Infer method) (push.push.PusH method) p_wait() (push.push.PusH method) params (messages.ReceiveGetAckMSG attribute) (messages.ReceiveParametersAckPDMSG attribute) params_grad (messages.ReceiveGetAckMSG attribute) Particle (class in push.particle) particle_ids() (push.particle.Particle method) (push.push.PusH method) particle_to_device (messages.NELBroadcastParticlesMSG attribute) ParticleCache (class in context_switch) ParticleCacheLRU (class in context_switch) particles() (context_switch.ParticleCache method) (context_switch.ParticleCacheLRU method) (node_event_loop.NodeEventLoop method) ParticleView (class in push.particle) PFuture (class in push.pfuture) pid (messages.ReceiveFuncMSG attribute) (messages.ReceiveGetAckMSG attribute) (messages.ReceiveGetMSG attribute) (messages.ReceiveParametersPDMSG attribute) (messages.ReceiveParticleInitPDMSG attribute) (messages.ReceiveRegisterPDMSG attribute) pid_caller (messages.ReceiveGetMSG attribute) pid_fid (messages.NELSaveModel attribute) (messages.NELSaveModelAckPDMSG attribute) (messages.ReceiveFuncAckPDMSG attribute) (messages.ReceiveFuncMSG attribute) (messages.ReceiveFuncPDMSG attribute) (messages.ReceiveGetMSG attribute) (messages.ReceiveParametersAckPDMSG attribute) (messages.ReceiveParametersPDMSG attribute)

pid_to (messages.ReceiveFuncPDMSG attribute) posterior_pred() (push.bayes.ensemble.Ensemble method) (push.bayes.infer.Infer method) (push.bayes.stein_vgd.SteinVGD method) (push.bayes.swag.MultiSWAG method) PusH (class in push.push) push.bayes.ensemble module push.bayes.infer module push.bayes.stein_vgd module push.bayes.swag module push.bayes.utils module push.particle module push.pfuture module push.push module

R

read() (context_switch.ParticleCacheLRU method) receive (messages.ReceiveParticleInitPDMSG attribute) ReceiveFuncAckMSG (class in messages) ReceiveFuncAckPDMSG (class in messages) ReceiveFuncMSG (class in messages) ReceiveFuncPDMSG (class in messages) ReceiveGetAckMSG (class in messages) ReceiveGetMSG (class in messages) ReceiveParametersAckPDMSG (class in messages)

ReceiveParametersPDMSG (class in messages) ReceiveParticleInitAckPDMSG (class in messages) ReceiveParticleInitPDMSG (class in messages) ReceiveRegisterAckPDMSG (class in messages) ReceiveRegisterPDMSG (class in messages) register_receive() (node_event_loop.NodeEventLoop method) (push.particle.Particle method) release() (context_switch.ParticleCache method) result (messages.ReceiveFuncAckPDMSG attribute)

S

save() (push.push.PusH method) save_to_disk() (context_switch.ParticleCache method) scheduler_step() (node_event_loop.NodeEventLoop method) (push.particle.Particle method) send() (node_event_loop.NodeEventLoop method) (push.particle.Particle method)

state (messages.ReceiveParticleInitPDMSG attribute) (messages.ReceiveRegisterPDMSG attribute) SteinVGD (class in push.bayes.stein_vgd) step() (node_event_loop.NodeEventLoop method) (push.particle.Particle method)

T

to_device() (in module utils) torch_squared_exp_kernel() (in module push.bayes.stein_vgd) torch_squared_exp_kernel_grad() (in module push.bayes.stein_vgd) train_deep_ensemble() (in module push.bayes.ensemble)

train_mswag() (in module push.bayes.swag) train_svgd() (in module push.bayes.stein_vgd) try_pin() (context_switch.ParticleCache method) try_read() (context_switch.ParticleCache method)

U

unflatten_like() (in module push.bayes.utils) unpin() (context_switch.ParticleCache method)

update_theta() (in module push.bayes.swag) utils module

V

view() (push.particle.ParticleView method)

W

wait() (node_event_loop.NodeEventLoop method) (push.particle.Particle method) (push.pfuture.PFuture method)

waitable module Waitable (class in waitable) write() (context_switch.ParticleCacheLRU method)

Z

zero_grad() (node_event_loop.NodeEventLoop method) (push.particle.Particle method)