You helped with type annotations using the jaxtyping library

💬 3+ Chats

AI tool for ML research with focus on precise math, Jax/Flax code, and accurate knowledge use

💬 10+ Chats