Back to browse
GitHub Repository

High-performance, branchless numerical stability kernels and compiler-optimized core infrastructure for advanced JAX/XLA deep learning architectures.

0 starsPython

AI-related Jax module (I hate if)

by PJHkorea·Jul 6, 2026·2 points·4 comments

Similar Projects

Developer Tools●●●Banger

Interactive fluid simulation in Jax using Brinkman penalization

Draw obstacles with your mouse and get gradient-based inverse design without adjoint solvers.

WizardrySolve My ProblemNiche Gem
arriemeijer
201mo ago