published Published about 1 month ago
OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX
Python
published Published about 2 months ago
BordAX: A High-Performance JAX Framework for Programmatic Reinforcement Learning
Python
published Published 3 months ago
Cosmologix: Fast, accurate and differentiable distances in the universe with JAX
Python
published Published 3 months ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs
Python Cuda
published Published 6 months ago

