KPPax compiles KPP chemistry mechanisms into differentiable JAX code
On POLLU, online fine-tuning cuts biogenic emission-correction error by a factor of 2.4 to 5.
Stories per day · 30 days
1 RSS
6 stories, newest first · one card per story
Stories tagged #jax
Simulation-Free Learning of Population Dynamics with Wasserstein Lagrangian Residuals
Double-Stitch learns Wasserstein Lagrangian population dynamics simulation-free, training 4-14x faster than solver-based WLM while covering periodic dynamics.
A Coding Guide to Google Research’s Kauldron: Configs That Are Plain Data, Components Wired by String, and a JAX Trainer You Can Read End to End
A coding guide demonstrates Google Research's Kauldron, a modular JAX trainer using plain-data configs and string wiring.
E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU
Open-source e3j speeds Euclidean-equivariant GPU and TPU kernels, beating cuEquivariance on a MACE simulation.
Sakana AI Researchers Introduce PC-ALM, a Layer-Local Alternative to Backpropagation That Trains 1000-Layer Networks
Sakana AI's PC-ALM adds per-layer Lagrange multipliers to predictive coding, matching backprop on networks up to 1000 layers with layer-local updates.
6
Hierarchical NeRF with JAX3D for Volumetric Rendering, Novel-View Synthesis, and 3D Reconstruction
MarkTechPost tutorial implements a hierarchical NeRF in JAX using jax3d volume-rendering primitives for novel-view synthesis and 3D reconstruction.
2