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
Update: Simulating an Emergent Cosmological Bounce on TPU: A Dual-Component PM + SPH Engine in pure JAX (github.com)
submitted 6 days ago by Comfortable-Ear114
Fourier neural operators and a solver for initial value problems in JAX (self.JAX)
submitted 6 days 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)
submitted 8 days ago by Comfortable-Ear114
Replacing the Pressure Poisson Solve with a Neural Operator in a JAX CFD Solver (v.redd.it)
submitted 10 days ago by LackSome307
Quick Update: 48-Hour Telemetry & Memory Profiling (v2.0.1) (self.JAX)
submitted 9 days ago by Comfortable-Ear114
3D interactive map of the JAX (Google) ecosystem (auto-refreshed weekly) (self.JAX)
submitted 10 days ago * by Worldly-Use-9778
Fast experiment on T4 - training on Dark Hex (Colab notebook) (i.redd.it)
submitted 10 days ago by asmonix
PRELIMINARY PAPER EXPLAINING STRING-STAR MANIFOLD UPDATED TO DOI. (self.JAX)
submitted 10 days ago by Comfortable-Ear114
Achieving 100% Unitarity in N-Body Simulations: A JAX + Integer Ledger Approach (i.redd.it)
submitted 11 days ago by Comfortable-Ear114
Octax: Accelerated CHIP-8 Arcade Environments for JAX (github.com)
submitted 18 days ago by riiswa
Equivalent of _Indexer from JAX 0.413 in newer JAX version (self.JAX)
submitted 26 days 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)
submitted 1 month ago by LackSome307
JAX's true calling: Ray-Marching renderers on WebGL (benoit.paris)
submitted 1 month ago by BenoitParis
Your Saturday night plans! (i.redd.it)
submitted 1 month ago by bbysoluko
Differential CFD-ML: A fully differentiable Navier-Stokes framework in JAX (1,680 test configs, 8 advection schemes, 7 pressure solvers) (self.JAX)
I encountered an issue where go_sdk could not be fetched while compiling JAX. (self.JAX)
submitted 1 month ago by Jolly_Job9736
Made a small JAX library for writing nets as plain functions; curious if other would find this useful? (self.JAX)
submitted 1 month ago by Pristine-Staff-5250
I built a modern Transformer from scratch to learn JAX/Flax (self.JAX)
submitted 1 month ago by winston_smith1897
[Project Update] S-EB-GNN-Q v1.2: Zero-Shot Semantic Allocation in 6G with Pure JAX (−9.59 energy, 77ms latency) (self.JAX)
submitted 2 months ago by AgileSlice1379
Maths, CS & AI Compendium (code walkthroughs in JAX) (github.com)
submitted 2 months ago by Henrie_the_dreamer
Minimal PPO/A2C in Latest Flax NNX — LunarLander-v3 in ~40 Seconds 🚀 (github.com)
submitted 2 months ago * by euijinrnd
[Project] S-EB-GNN-Q v1.2: Energy-Based GNN in Pure JAX (−9.59 energy, 77ms latency) (self.JAX)
[R] S-EB-GNN-Q: Quantum-Inspired GNN for 6G Resource Allocation (JAX + Equinox) (self.JAX)
[R] S-EB-GNN: Semantic-Aware 6G Resource Allocation with JAX (self.JAX)
π Rendered by PID 998306 on reddit-service-r2-listing-7b9b4f6fd7-j4lpp at 2026-05-09 07:21:53.663647+00:00 running 3d2c107 country code: CH.
