[D] Any less-boilerplate framework for Jax/Flax/Haiku? | allainews.com

Sept. 30, 2022, 2:57 a.m. | /u/KingsmanVince

Machine Learning www.reddit.com

I have been looking for a framework/library that can be like/smiliar to [PyTorch Lightning](https://www.pytorchlightning.ai/). I even checked on [Awesome-Jax](https://github.com/n2cholas/awesome-jax). Do we have any framework but for Jax/flax/haiku?

I mostly need features: Checkpoint saving, reproducibility, and logging.

framework haiku jax machinelearning

More from www.reddit.com / Machine Learning

[R] Training-free Graph Neural Networks and the Power of Labels as Features 10 hours ago | www.reddit.com

features free graph graph neural networks +6

[D] Modern best coding practices for Pytorch (for research)? 13 hours ago | www.reddit.com

coding config example good +14

[R] Is Model Collapse Inevitable? Breaking the Curse of Recursion by Accumulating Real and Synthetic … 15 hours ago | www.reddit.com

breaking data machinelearning model collapse +3

[P] I reproduced Anthropic's recent interpretability research 16 hours ago | www.reddit.com

anthropic attention basic capabilities +8

[R] KAN: Kolmogorov-Arnold Networks 17 hours ago | www.reddit.com

abstract every function functions +11

[D] Looking for a recent study/paper/article that showed that an alternate model with a similar … 17 hours ago | www.reddit.com

article conversation machinelearning nothing +4

[2404.10667] VASA-1: Lifelike Audio-Driven Talking Faces Generated in Real Time 18 hours ago | www.reddit.com

audio generated machinelearning vasa +1

[D] Is RPE still a valid approach, or is RoPE entirely superior? 22 hours ago | www.reddit.com

attention datasets embed information +8

[D] TensorDock — GPU Cloud Marketplace, H100s from $2.49/hr 1 day ago | www.reddit.com

building cloud cloud gpu gpu +17

Data Architect

@ University of Texas at Austin | Austin, TX

View on ai-jobs.net

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

View on ai-jobs.net

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

View on ai-jobs.net

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

View on ai-jobs.net

C003549 Data Analyst (NS) - MON 13 May

@ EMW, Inc. | Braine-l'Alleud, Wallonia, Belgium

View on ai-jobs.net

Marketing Decision Scientist

@ Meta | Menlo Park, CA | New York City

View on ai-jobs.net