Alternatives and similar repositories for doxa

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

• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆248Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆106Updated 7 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• google-deepmind / chex
  ☆865Updated 2 weeks ago
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆299Updated last year
• google-deepmind / distrax
  ☆581Updated 2 weeks ago
• GFNOrg / gflownet
  Generative Flow Networks
  ☆643Updated 2 years ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆350Updated last month
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆473Updated 3 years ago
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆274Updated this week
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆420Updated 4 months ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆254Updated 4 months ago
• alexhernandezgarcia / gflownet
  Generative Flow Networks - GFlowNet
  ☆262Updated last week
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆268Updated 2 years ago
• google / grain
  Library for reading and processing ML training data.
  ☆478Updated this week
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆386Updated 3 months ago
• google-deepmind / jaxline
  ☆157Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆91Updated last month
• google-deepmind / tf2jax
  ☆115Updated this week
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆406Updated this week
• google-deepmind / enn
  ☆312Updated 4 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆206Updated 5 months ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆281Updated last week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆262Updated last week
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆140Updated last year
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆176Updated last year
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆131Updated last year
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆981Updated 3 months ago
• lucidrains / equiformer-pytorch
  Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…
  ☆273Updated 7 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆345Updated 3 months ago