BitTransformerLM/example.py

from bit_transformer import example_training_step

if __name__ == "__main__":
    loss, telemetry = example_training_step()
    print("Training loss:", loss)
    print("Available telemetry:", list(telemetry.keys()))