Alternatives and similar repositories for DLO-JZ

Users that are interested in DLO-JZ are comparing it to the libraries listed below

• PrincetonUniversity / multi_gpu_training
  ☆370Updated 4 months ago
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆512Updated last week
• nikhilvyas / SOAP
  ☆238Updated last year
• pasteurlabs / tesseract-core
  Universal, autodiff-native software components for Simulation Intelligence. 📦
  ☆91Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆366Updated last week
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆310Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆311Updated 3 months ago
• karimknaebel / turm
  TUI for the Slurm Workload Manager
  ☆419Updated last week
• jax-ml / jax-ai-stack
  ☆263Updated this week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆119Updated last year
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆70Updated 8 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆506Updated 6 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆98Updated last month
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆381Updated this week
• NERSC / sc23-dl-tutorial
  SC23 Deep Learning at Scale Tutorial Material
  ☆49Updated last year
• google-deepmind / nanodo
  ☆289Updated last year
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆184Updated last year
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆437Updated last month
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆478Updated last week
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆558Updated 3 weeks ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆406Updated this week
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆41Updated 2 years ago
• NERSC / sc24-dl-tutorial
  SC24 Deep Learning at Scale Tutorial Material
  ☆33Updated 11 months ago
• google-research / jaxpruner
  ☆234Updated 11 months ago
• jean-zay-users / jean-zay-doc
  Collaborative documentation for and from Jean Zay users. Official Jean Zay documentation: http://www.idris.fr/eng/jean-zay/
  ☆130Updated 2 months ago
• google-research / spherical-cnn
  ☆130Updated last week
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆110Updated 3 months ago
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆507Updated this week
• aai-institute / continuiti
  Learning function operators with neural networks.
  ☆35Updated last year
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆236Updated 2 weeks ago