AI Agent Skills

Solve puzzles. Learn CUDA.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Puzzles for learning Triton

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Run Keras models in the browser, with GPU support using WebGL

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT a…