JAX, Flax, Optax, and Equinox patterns for ML training. Covers JIT, vmap, pmap, TPU usage, and functional model design. Use when building or debugging JAX-based training pipelines.
Use this skill when the user asks to save, remember, recall, or organize memories. Triggers on: 'remember this', 'save t...
Your own personal AI assistant. Any OS. Any Platform. The lobster way. 🦞