JAX

Timeline

1. Oct 8, 2026 · Research paper · 1 source
   asdex: Automatic Sparse Differentiation in JAX
   Automatic sparse differentiation (ASD) exploits this structure in four steps: detection of the input-agnostic sparsity pattern, coloring of a graph to group columns or rows that can share an AD pass, compressed differentiation to compute a compressed derivative matrix with one AD pass per color, and finally decompression into the original sparsity pattern.
2. Oct 6, 2026 · Research paper · 1 source
   UWB Meets Crazyflow: Simulating Degraded Feedback at Scale for Aerial Robotics
   In this work, we introduce Crazyflow, an accurate, differentiable simulator built on JAX.
3. Oct 6, 2026 · Research paper · 1 source
   Physics-based Sphere Packing for Lagrangian Mesh Morphing
   We present JamTet, a physics-based sphere-packing framework for volumetric meshing and morphing.
4. Oct 6, 2026 · Research paper · 1 source
   BluffJAX: Adversarial Imperfect Information Games in JAX
   We introduce BluffJAX: an open-source suite of adversarial imperfect information games in JAX.
5. Jun 5, 2024 · Open-source release · 1 source
   pytorch/pytorch v2.3.1: PyTorch 2.3.1 Release, bug fix release
   This release is meant to fix the following issues (regressions / silent correctness):

Often appears with