File size: 1,100 Bytes
6b89792
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
LICENSE
NOTICE
README.md
setup.py
se3_transformer/__init__.py
se3_transformer.egg-info/PKG-INFO
se3_transformer.egg-info/SOURCES.txt
se3_transformer.egg-info/dependency_links.txt
se3_transformer.egg-info/top_level.txt
se3_transformer/data_loading/__init__.py
se3_transformer/data_loading/data_module.py
se3_transformer/data_loading/qm9.py
se3_transformer/model/__init__.py
se3_transformer/model/basis.py
se3_transformer/model/fiber.py
se3_transformer/model/transformer.py
se3_transformer/model/layers/__init__.py
se3_transformer/model/layers/attention.py
se3_transformer/model/layers/convolution.py
se3_transformer/model/layers/linear.py
se3_transformer/model/layers/norm.py
se3_transformer/model/layers/pooling.py
se3_transformer/runtime/__init__.py
se3_transformer/runtime/arguments.py
se3_transformer/runtime/callbacks.py
se3_transformer/runtime/gpu_affinity.py
se3_transformer/runtime/inference.py
se3_transformer/runtime/loggers.py
se3_transformer/runtime/metrics.py
se3_transformer/runtime/training.py
se3_transformer/runtime/utils.py
tests/__init__.py
tests/test_equivariance.py
tests/utils.py