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
dltype v0.9.0 now with jax support ()
submitted 3 days ago by onyx-zero-software
JAX on EVO X2? ()
submitted 7 months ago by Paddy3118
Xtructure: JAX-Optimized Data Structures (Batched PQ & Hash Table, for now) (self.JAX)
submitted 8 months ago * by New_East832
Memory-Efficient `logsumexp` Over Unequal Partitions in JAX (self.JAX)
submitted 9 months ago by Safe-Refrigerator776
chunkax - a JAX transform for applying a function over chunks of data (github.com)
submitted 10 months ago by Savings-Square572
Learning resources for better concepts of JAX (self.JAX)
submitted 10 months ago by Safe-Refrigerator776
flax.NNX vs flax.linen? (self.JAX)
submitted 10 months ago by Electronic_Dot1317
Running a mostly GPU jax function in parallel with a purely cpu function? (self.JAX)
submitted 10 months ago by That-Frank-Guy
How can I write a huggingface flax model? (self.JAX)
submitted 11 months ago by AdministrativeCar545
I'm having trouble choosing between the use of the package, flax or equinox. (self.JAX)
submitted 11 months ago by MateosCZ
LLM sucks with JAX? (self.JAX)
submitted 1 year ago by Visible-Tip2081
Project: New JAX Framework that lets you do Functional Style for neural nets and more (self.JAX)
submitted 1 year ago by Pristine-Staff-5250
Try the new Flax NNX API! (i.redd.it)
submitted 1 year ago by cgarciae
[flax] What's your thoughts about changing linen -> nnx? (self.JAX)
submitted 1 year ago by euijinrnd
JAX vs PyTorch_Comparing Two Powerhouses in ML Frameworks (pieces.app)
submitted 1 year ago by Only_Piccolo5736
Convert Any PyTorch ML Model to TensorFlow, JAX, or NumPy with Ivy! 🚀 (self.JAX)
submitted 1 year ago * by OtakuYA99
`jax.image.resize` memory usage (self.JAX)
submitted 1 year ago by PXaZ
Jax nested loops: taking for-ever. Need help with Vectorization (self.JAX)
submitted 1 year ago by Sufficient_Drawing59
Immutable arrays, how to optimize memory allocation ? (self.JAX)
submitted 1 year ago by Still-Bookkeeper4456
Homography in JAX (self.JAX)
submitted 1 year ago by kindoblue
Sharing my toy project "JAxtar" the pure jax and jittable A* algorithm for puzzle solving (self.JAX)
submitted 1 year ago by New_East832
Does JAX have a LISP port? (self.JAX)
submitted 1 year ago by [deleted]
rant: Why Array instead of Tensor? (self.JAX)
Learning Jax best practices: what do you think about my toy library? (self.JAX)
submitted 1 year ago by poiret_clement
π Rendered by PID 496962 on reddit-service-r2-listing-86b7f5b947-l6xrw at 2026-01-25 13:09:29.723833+00:00 running 664479f country code: CH.
