Alternatives and similar repositories for yawc

Users that are interested in yawc are comparing it to the libraries listed below

• kvfrans / splus
  ☆119Updated 5 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆96Updated 5 months ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆50Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆110Updated 3 weeks ago
• google-research / jaxpruner
  ☆234Updated 9 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 11 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• davisyoshida / qax
  If it quacks like a tensor...
  ☆59Updated last year
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆118Updated this week
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆131Updated last year
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆33Updated 2 years ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆47Updated 9 months ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆40Updated 2 years ago
• TheodoreWolf / hyperoptax
  Parallel hyperparameter tuning with JAX
  ☆36Updated 3 months ago
• kabouzeid / turm
  TUI for the Slurm Workload Manager
  ☆224Updated 2 months ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆69Updated 6 months ago
• shikaiqiu / compute-better-spent
  ☆61Updated last year
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆190Updated 11 months ago
• j-towns / scanagram
  Tidy autoregressive inference in JAX
  ☆15Updated 2 months ago
• CLAIRE-Labo / python-ml-research-template
  A template for starting reproducible Python machine-learning projects with hardware acceleration. Find an example at https://github.com/C…
  ☆109Updated 5 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆142Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆358Updated 4 months ago
• marin-community / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆212Updated last week
• j10labs / wandview
  Mobile Viewer for W&B, built on top of Flutter.
  ☆38Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆303Updated 2 months ago
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆116Updated last month
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆253Updated 2 years ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• Artur-Galstyan / statedict2pytree
  ☆44Updated 2 weeks ago