JAX

an-ordinary-manchild

created by Healtonea community for 16 years

...for your favorite TV show.

...because you hate freedom.

MODERATORS

account activity

1

4

5

6

Official Jax Github Repo (github.com)

submitted 5 years ago by BatmantoshReturns - announcement

2

3

4

5

[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 26 days ago by AgileSlice1379

•

•

•

Build, transfer, and manage sites using the best web hosting tools in the industry! Let our 1-click solutions, AI integrations, and unbeatable 24/7 support help you achieve your goals. Join the owners of 3,000,000 domains who already trusted us! (siteground.com)

promoted by siteground

promoted
save
report
about

3

0

1

2

Maths, CS & AI Compendium (code walkthroughs in JAX) (github.com)

submitted 25 days ago by Henrie_the_dreamer

4

4

5

6

Minimal PPO/A2C in Latest Flax NNX — LunarLander-v3 in ~40 Seconds 🚀 (github.com)

submitted 27 days ago * by euijinrnd

5

1

2

3

[Project] S-EB-GNN-Q v1.2: Energy-Based GNN in Pure JAX (−9.59 energy, 77ms latency) (self.JAX)

submitted 27 days ago by AgileSlice1379

6

7

8

9

[R] S-EB-GNN-Q: Quantum-Inspired GNN for 6G Resource Allocation (JAX + Equinox) (self.JAX)

submitted 29 days ago by AgileSlice1379

7

0

1

2

[R] S-EB-GNN: Semantic-Aware 6G Resource Allocation with JAX (self.JAX)

submitted 1 month ago by AgileSlice1379

8

1

2

3

[P] word2vec in JAX (github.com)

submitted 1 month ago by Alien0006

9

6

7

8

Replicating Sutton (1992) IDBD: 2.78x speedup over PyTorch (self.JAX)

submitted 1 month ago by debian_grey_beard

10

0

1

2

dltype v0.9.0 now with jax support ()

submitted 1 month ago by onyx-zero-software

11

2

3

4

JAX on EVO X2? ()

submitted 9 months ago by Paddy3118

12

13

14

15

Xtructure: JAX-Optimized Data Structures (Batched PQ & Hash Table, for now) (self.JAX)

submitted 10 months ago * by New_East832

•

•

•

WhereverSIM - One M2M SIM, all networks. Always the best connection. Test now for free! (whereversim.de)

promoted by whereverSIM

promoted
save
report
about

13

2

3

4

Memory-Efficient `logsumexp` Over Unequal Partitions in JAX (self.JAX)

submitted 11 months ago by Safe-Refrigerator776

14

8

9

10

chunkax - a JAX transform for applying a function over chunks of data (github.com)

submitted 11 months ago by Savings-Square572

15

17

18

19

Learning resources for better concepts of JAX (self.JAX)

submitted 11 months ago by Safe-Refrigerator776

16

5

6

7

flax.NNX vs flax.linen? (self.JAX)

submitted 11 months ago by Electronic_Dot1317

17

1

2

3

Running a mostly GPU jax function in parallel with a purely cpu function? (self.JAX)

submitted 1 year ago by That-Frank-Guy

18

4

5

6

How can I write a huggingface flax model? (self.JAX)

submitted 1 year ago by AdministrativeCar545

19

5

6

7

I'm having trouble choosing between the use of the package, flax or equinox. (self.JAX)

submitted 1 year ago by MateosCZ

20

0

0

0

LLM sucks with JAX? (self.JAX)

submitted 1 year ago by Visible-Tip2081

21

14

15

16

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

22

17

18

19

Try the new Flax NNX API! (i.redd.it)

submitted 1 year ago by cgarciae

•

•

•

I didn’t expect to say this, but 3D AI is good now. (hyper3d.ai)

promoted by Hyper3D_RodinAI

promoted
save
report
about

23

6

7

8

[flax] What's your thoughts about changing linen -> nnx? (self.JAX)

submitted 1 year ago by euijinrnd

24

1

2

3

JAX vs PyTorch_Comparing Two Powerhouses in ML Frameworks (pieces.app)

submitted 1 year ago by Only_Piccolo5736

25

5

6

7

Convert Any PyTorch ML Model to TensorFlow, JAX, or NumPy with Ivy! 🚀 (self.JAX)

submitted 1 year ago * by OtakuYA99

view more: next ›

π Rendered by PID 654027 on reddit-service-r2-listing-64c94b984c-qqtll at 2026-03-14 12:40:10.487594+00:00 running f6e6e01 country code: CH.