Alternatives and similar repositories for DiffRast:

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

• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆60Updated 7 months ago
• cloneofsimo / zeroshampoo
  ☆33Updated 6 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆33Updated 7 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆39Updated 2 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 8 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 months ago
• packquickly / schedule_free_optx
  Schedule free optimiser implemented in JAX using Optimistix
  ☆14Updated 9 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆85Updated 7 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 7 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆53Updated 7 months ago
• iree-org / iree-jax
  ☆51Updated 7 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• znah / zff
  8-bit computational substrates
  ☆39Updated 8 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 4 months ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆28Updated 2 months ago
• ethansmith2000 / TransformerExperiments
  ☆19Updated 5 months ago
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆57Updated 9 months ago
• Z-T-WANG / LaProp-Optimizer
  Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"
  ☆27Updated 4 years ago
• DAIOS-AI / mindscript
  A programming language for formal/informal computation.
  ☆41Updated this week
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆29Updated 4 months ago
• Silent-Zebra / twisted-smc-lm
  ☆24Updated 2 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆35Updated last month