The Agent Skills Directory

[SAFE]: The skill is a high-performance deep learning extension for JAX, authored by NVIDIA. It provides core primitives and Flax NNX layers for equivariant neural networks.
[COMMAND_EXECUTION]: The file triangle/triton_utils.py uses subprocess.run to check the version of the ptxas compiler. This is a benign environment check used to adjust kernel compilation parameters based on the installed CUDA toolkit version.
[EXTERNAL_DOWNLOADS]: The skill relies on well-known and trusted ecosystem packages including jax, flax, and triton. It references official documentation and source repositories for these tools.

cuequivariance-jax