Alternatives and similar repositories for doxa

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

• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆118Updated last year
• google / learned_optimization
  ☆792Updated last month
• lucidrains / En-transformer
  Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network
  ☆226Updated last year
• lucidrains / equiformer-pytorch
  Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…
  ☆285Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆258Updated 2 years ago
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆307Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆363Updated 2 weeks ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated 3 weeks ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆475Updated 3 years ago
• GFNOrg / gflownet
  Generative Flow Networks
  ☆663Updated 2 years ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆141Updated 2 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 3 years ago
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆303Updated 2 weeks ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• google / grain
  Library for reading and processing ML training data.
  ☆643Updated last week
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆431Updated 10 months ago
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆456Updated 9 months ago
• google-deepmind / chex
  ☆913Updated 2 weeks ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆160Updated 3 years ago
• recursionpharma / gflownet
  GFlowNet library specialized for graph & molecular data
  ☆280Updated 7 months ago
• google-deepmind / distrax
  ☆613Updated last month
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆280Updated 2 years ago
• chaitjo / geometric-gnn-dojo
  Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks (ICML …
  ☆518Updated 3 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated 2 years ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆372Updated 2 years ago
• ctlllll / SGConv
  ☆164Updated 2 years ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆311Updated 6 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆45Updated last year
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆307Updated last year