xla
f4a612cb - Training example for llama3; misc changes to make the training work. (#7194)

Commit
1 year ago
Training example for llama3; misc changes to make the training work. (#7194)
Author
Parents
  • experimental/torch_xla2
    • docs
      • File
        torch_xla2_dynamo.md
    • examples
      • File
        __init__.py
      • File
        _grad_of_attention.py
      • train_llama
        • File
          README.md
        • File
          __init__.py
        • File
          model.py
        • File
          train_llama_lightning.py
        • File
          utils.py
    • test
      • File
        test_interop.py
    • torch_xla2
      • File
        config.py
      • File
        interop.py
      • ops
        • File
          jaten.py
        • File
          jtorch.py
      • File
        tensor.py