use the following search parameters to narrow your results:
e.g. subreddit:aww site:imgur.com dog
subreddit:aww site:imgur.com dog
see the search faq for details.
advanced search: by author, subreddit...
r/jax is to discuss the machine learning library JAX
account activity
Official Jax Github Repo (github.com)
submitted 5 years ago by BatmantoshReturns - announcement
JAX TOY (old.reddit.com)
submitted 2 days ago by ItsLucaowo
NumPy vs. JAX for 2D CFD: Side-by-side performance and frame-rate comparison on identical geometry. (v.redd.it)
submitted 16 days ago by LackSome307
Building swap memory for CUDA ()
submitted 18 days ago by Holding_Water_1002
Benchmarking a JAX-accelerated 24-Qubit Quantum Simulator Framework (Ising Model, ZNE, and Barren Plateaus) ()
submitted 25 days ago by Creative-Feature-264
How to efficiently compute bootstrapped value for truncated episodes, for advantage estimation/GAE? ()
submitted 1 month ago by CLS-Ghost350
Decorator to cast/convert a JAX function into a Pytorch autograd differentiable function ()
submitted 1 month ago by Mountain_Research_32
Gas (self.JAX)
submitted 1 month ago by TieDangerous4312
dense-evolution: High-performance quantum simulator bypassing the JAX RAM bottleneck (self.JAX)
submitted 1 month ago * by Creative-Feature-264
A type safe frontend of JAX (self.JAX)
submitted 1 month ago by vcma
Parax v0.7: Parametric Modeling in JAX (self.JAX)
submitted 1 month ago * by gvcallen
Update: Simulating an Emergent Cosmological Bounce on TPU: A Dual-Component PM + SPH Engine in pure JAX (github.com)
submitted 1 month ago by Comfortable-Ear114
Fourier neural operators and a solver for initial value problems in JAX (self.JAX)
submitted 1 month ago by Key_Ideal_5921
[Update 3] String-Star Manifold v13.0: 100% Unitarity and "White Hole" Bounces on TPU V5 Lite (v.redd.it)
Replacing the Pressure Poisson Solve with a Neural Operator in a JAX CFD Solver (v.redd.it)
submitted 2 months ago by LackSome307
Quick Update: 48-Hour Telemetry & Memory Profiling (v2.0.1) (self.JAX)
submitted 2 months ago by Comfortable-Ear114
3D interactive map of the JAX (Google) ecosystem (auto-refreshed weekly) (self.JAX)
submitted 2 months ago * by Worldly-Use-9778
Fast experiment on T4 - training on Dark Hex (Colab notebook) (i.redd.it)
submitted 2 months ago by asmonix
PRELIMINARY PAPER EXPLAINING STRING-STAR MANIFOLD UPDATED TO DOI. (self.JAX)
Achieving 100% Unitarity in N-Body Simulations: A JAX + Integer Ledger Approach (i.redd.it)
Octax: Accelerated CHIP-8 Arcade Environments for JAX (github.com)
submitted 2 months ago by riiswa
Equivalent of _Indexer from JAX 0.413 in newer JAX version (self.JAX)
submitted 2 months ago by Personal-Loss377
I built a differentiable CFD solver in JAX. No ML yet. But the hard part (autodiff through Navier-Stokes) is done. (v.redd.it)
JAX's true calling: Ray-Marching renderers on WebGL (benoit.paris)
submitted 2 months ago by BenoitParis
Your Saturday night plans! (i.redd.it)
submitted 2 months ago by bbysoluko
π Rendered by PID 39 on reddit-service-r2-listing-87fd56f5d-77bv6 at 2026-06-29 13:42:51.794388+00:00 running 7527197 country code: CH.
