Remix.run Logo
Show HN: AI-related Jax module (I hate if)(github.com)
3 points by PJHkorea 13 hours ago | 5 comments
Hussain04 13 hours ago | parent | next [-]

For optimizers.py, are you relying purely on boolean arithmetic for these masks, or did you find a more elegant mathematical workaround to avoid the branching overhead entirely?

PJHkorea 11 hours ago | parent | next [-]

Thanks to that, I discovered additional errors and corrected them in various ways. Thank you.

Hussain04 10 hours ago | parent [-]

Good to hear!

PJHkorea 12 hours ago | parent | prev [-]

i naver imagined it i've made the corrections thank you!

PJHkorea 13 hours ago | parent | prev [-]

optimizers.py -> using algebraic expressing without using 'if'

geometry.py -> I leveled it out

math_guardrails.py -> facilitates learning through slight tiliing

jax it truly precious it cures my insomnia