batch_invariant_kernel

To try out the example of the code by running the following commands:

uv run https://huggingface.co/gagan3012/batch_invariant_kernel/resolve/main/readme_example.py

Credits

Work here is based on the following repositories:

• Batch Invariant Kernel by Thinking Machines Lab