Sven Mika cf21c634a3 [RLlib] Fix deprecated warning for torch_ops.py (soft-replaced by torch_utils.py). (#19982) 3 years ago
..
__init__.py f77c638d6d Pytorch AttentionNet (#9088) 4 years ago
convtranspose2d_stack.py 8726521604 [RLlib] JAXPolicy prep PR #2 (move get_activation_fn (backward-compatibly), minor fixes and preparations). (#13091) 3 years ago
gru_gate.py d5604eaba3 [RLlib] Attention nets PyTorch support and cleanup (using traj. view API). (#12029) 3 years ago
multi_head_attention.py cf21c634a3 [RLlib] Fix deprecated warning for torch_ops.py (soft-replaced by torch_utils.py). (#19982) 3 years ago
noisy_layer.py 8726521604 [RLlib] JAXPolicy prep PR #2 (move get_activation_fn (backward-compatibly), minor fixes and preparations). (#13091) 3 years ago
relative_multi_head_attention.py cf21c634a3 [RLlib] Fix deprecated warning for torch_ops.py (soft-replaced by torch_utils.py). (#19982) 3 years ago
skip_connection.py b2bcab711d [RLlib] Attention Nets: tf (#12753) 3 years ago