会员服务
JZ-Tree: GPU friendly neighbour search and friends-of-friends with dual tree walks in JAX plus CUDA
Arxiv
0+阅读 · 4月7日
Fast-Vollib: A Fast Implied Volatility Library for Pythonwith PyTorch, JAX, and CUDA Fused-Kernel Backends
Arxiv
0+阅读 · 4月29日
jaxsgp4: GPU-accelerated mega-constellation propagation with batch parallelism
Arxiv
0+阅读 · 3月29日
JAX-BEM: Gradient-Based Acoustic Shape Optimisation via a Differentiable Boundary Element Method
Arxiv
0+阅读 · 4月23日
Self-Scaled Broyden Family of Quasi-Newton Methods in JAX
Arxiv
0+阅读 · 3月11日
Compiler-First State Space Duality and Portable $O(1)$ Autoregressive Caching for Inference
Arxiv
0+阅读 · 3月10日
TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning
Arxiv
0+阅读 · 2月2日
KAGE-Bench: Fast Known-Axis Visual Generalization Evaluation for Reinforcement Learning
Arxiv
0+阅读 · 1月20日
JAXMg: A multi-GPU linear solver in JAX
Arxiv
0+阅读 · 1月20日
Efficient GPU-computing simulation platform JAX-PF for differentiable phase field model
Arxiv
0+阅读 · 2025年12月29日
Fast, Differentiable, GPU-Accelerated Ray Tracing for Multiple Diffraction and Reflection Paths
Arxiv
0+阅读 · 2025年12月31日
Differentiation Strategies for Acoustic Inverse Problems: Admittance Estimation and Shape Optimization
Arxiv
0+阅读 · 2025年11月14日
gfnx: Fast and Scalable Library for Generative Flow Networks in JAX
Arxiv
0+阅读 · 2025年11月20日
JAX-LaB: A High-Performance, Differentiable, Lattice Boltzmann Library for Modeling Multiphase Fluid Dynamics in Geosciences and Engineering
Arxiv
0+阅读 · 2025年11月15日
scipy.spatial.transform: Differentiable Framework-Agnostic 3D Transformations in Python
Arxiv
0+阅读 · 2025年11月22日
参考链接
微信扫码咨询专知VIP会员
Top