jax,
an archive of posts with this tag
| Jun 04, 2025 | The Jax adjoint scatter trick |
|---|---|
| Apr 01, 2025 | Why you should write Jax functions without broadcasting |
| Jan 17, 2025 | The mypy bug that just kept on giving |
an archive of posts with this tag
| Jun 04, 2025 | The Jax adjoint scatter trick |
|---|---|
| Apr 01, 2025 | Why you should write Jax functions without broadcasting |
| Jan 17, 2025 | The mypy bug that just kept on giving |