Alexander Nikulin

Results 3 repositories owned by Alexander Nikulin

faster-trajectory-transformer

99
Stars
12
Forks
Watchers

Implementation of Trajectory Transformer with attention caching and batched beam search

sac-n-jax

44
Stars
3
Forks
Watchers

Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch

prioritized_experience_replay

58
Stars
9
Forks
Watchers

Prioritized Experience Replay implementation with proportional prioritization