14 projects with the selected classifier
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
Context specific and dynamic gene regulatory network reconstruction and analysis
Symmetry Preserving Attention Networks
Accelerated gridworld navigation with JAX for deep reinforcement learning
Fast and differentiable acoustic simulations in JAX.
Acoustic lens design on GPUs with Automatic Differentiation.
NVIDIA cuTENSOR
A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
code and data for the Findings of ACL'23 paper Label Agnostic Pre-training for Zero-shot Text Classification
my test package
JAX-compatible Enumerations.
Supported by