LICENSE
README.md
pyproject.toml
relational_transformers/__init__.py
relational_transformers/batch.py
relational_transformers/checkpoints.py
relational_transformers/constants.py
relational_transformers/datasets.py
relational_transformers/evaluation.py
relational_transformers/losses.py
relational_transformers/model.py
relational_transformers/onnx.py
relational_transformers/py.typed
relational_transformers/torch_model.py
relational_transformers/training.py
relational_transformers.egg-info/PKG-INFO
relational_transformers.egg-info/SOURCES.txt
relational_transformers.egg-info/dependency_links.txt
relational_transformers.egg-info/requires.txt
relational_transformers.egg-info/top_level.txt
relational_transformers/backends/__init__.py
relational_transformers/backends/triton_data.py
relational_transformers/backends/triton_kernels.py
relational_transformers/backends/triton_model.py
tests/test_batch.py
tests/test_checkpoints.py
tests/test_cuda_triton.py
tests/test_datasets_losses_evaluation.py
tests/test_edge_contracts.py
tests/test_examples.py
tests/test_examples_end_to_end.py
tests/test_feature_heads.py
tests/test_high_level.py
tests/test_published_models.py
tests/test_realistic_onnx.py
tests/test_realistic_prediction.py
tests/test_realistic_training.py
tests/test_release_onnx_script.py
tests/test_torch_model.py
tests/test_triton_worklists.py
tests/test_validation.py