LICENSE
README.md
pyproject.toml
docs/gen_issue.py
examples/__init__.py
examples/_diffusion.py
examples/_grad_of_attention.py
examples/basic_training.py
examples/basic_training_jax.py
examples/eager_mode.py
examples/lightning_training.py
examples/torchbench_models/BERT_pytorch.py
examples/train_llama/__init__.py
examples/train_llama/model.py
examples/train_llama/train_llama_lightning.py
examples/train_llama/utils.py
hantoshi.egg-info/PKG-INFO
hantoshi.egg-info/SOURCES.txt
hantoshi.egg-info/dependency_links.txt
hantoshi.egg-info/requires.txt
hantoshi.egg-info/top_level.txt
test/__init__.py
test/test_base.py
test/test_context.py
test/test_conv.py
test/test_core_aten_ops.py
test/test_exports.py
test/test_functions.py
test/test_interop.py
test/test_libraries.py
test/test_mutations.py
test/test_ops.py
test/test_symbolic_shapes.py
test/test_unbounded_dynamism.py
test/gemma/__init__.py
test/gemma/config.py
test/gemma/model.py
test/gemma/test_gemma.py
test/gemma/tokenizer.py
test/llama/__init__.py
test/llama/llama_model.py
test/llama/model_exportable.py
test/llama/test_llama.py
test/moe/__init__.py
test/moe/model.py
test/moe/moe_test.py
test_dist/__init__.py
test_dist/test_distributed.py
torch_xla2/__init__.py
torch_xla2/config.py
torch_xla2/decompositions.py
torch_xla2/distributed.py
torch_xla2/environment.py
torch_xla2/export.py
torch_xla2/interop.py
torch_xla2/tensor.py
torch_xla2/tf_integration.py
torch_xla2/types.py
torch_xla2/ops/__init__.py
torch_xla2/ops/jaten.py
torch_xla2/ops/jc10d.py
torch_xla2/ops/jlibrary.py
torch_xla2/ops/jtorch.py
torch_xla2/ops/mappings.py
torch_xla2/ops/op_base.py
torch_xla2/ops/ops_registry.py