Alternatives and similar repositories for DLO-JZ

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

• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆65Updated last month
• NERSC / sc24-dl-tutorial
  SC24 Deep Learning at Scale Tutorial Material
  ☆32Updated 4 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆91Updated last year
• jax-ml / jax-ai-stack
  ☆179Updated last week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 5 months ago
• google-deepmind / tf2jax
  ☆114Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆246Updated 3 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆85Updated last week
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆39Updated 7 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/
  ☆122Updated 10 months ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆87Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆124Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆343Updated last month
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆173Updated 10 months ago
• aai-institute / continuiti
  Learning function operators with neural networks.
  ☆32Updated 9 months ago
• neuraloperator / bootcamp
  Bootcamp notebooks
  ☆54Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆121Updated last week
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆199Updated 4 months ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆387Updated this week
• google / grain
  Library for reading and processing ML training data.
  ☆449Updated this week
• nikhilvyas / SOAP
  ☆185Updated 6 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆423Updated last week
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆106Updated 10 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• srigas / jaxKAN
  Kolmogorov-Arnold Networks built on JAX
  ☆78Updated last month
• jax-ml / jax-llm-examples
  ☆118Updated 2 weeks ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆109Updated 3 weeks ago
• geometric-intelligence / TopoBench
  TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learning
  ☆120Updated this week
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆183Updated 8 months ago
• MashedP / dlip-pytorch-DL-good-practises
  DLiP course companion repository for practical 1
  ☆13Updated 4 months ago