Alternatives and similar repositories for DiffRast

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

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆115Updated last year
• spatialreasoners / spatialreasoners
  [CODEML @ ICML2025] Official implementation of the 🌀Spatial Reasoners python package
  ☆39Updated last month
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆243Updated 2 months ago
• TheStageAI / TorchIntegral
  Integral Neural Networks in PyTorch
  ☆127Updated last year
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated last year
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆71Updated last year
• iree-org / iree-jax
  ☆55Updated last year
• zaydzuhri / softpick-attention
  Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"
  ☆86Updated 4 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆352Updated 2 months ago
• kvfrans / splus
  ☆123Updated 7 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆306Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆138Updated last month
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆464Updated 9 months ago
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆129Updated 3 months ago
• graphcore-research / pytorch-tensor-tracker
  Flexibly track outputs and grad-outputs of torch.nn.Module.
  ☆13Updated 2 years ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆90Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆153Updated 11 months ago
• yilundu / ired_code_release
  ☆85Updated last year
• hyhieu / easy_pybind
  ☆32Updated last year
• facebookresearch / capi
  Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"
  ☆130Updated 3 weeks ago
• Francesco215 / autoregressive_diffusion
  Video Diffusion Model. Autoregressive, long context, efficient training and inference. WIP
  ☆34Updated 5 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆145Updated 3 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆63Updated 2 years ago
• Foundations-of-Computer-Vision / visionbook
  <Foundations of Computer Vision> Book
  ☆432Updated last month
• ESHyperscale / HyperscaleES
  Jax Codebase for Evolutionary Strategies at the Hyperscale
  ☆216Updated last month
• AllanYangZhou / universal_neural_functional
  ☆55Updated last year
• riverstone496 / awesome-second-order-optimization
  ☆28Updated 4 months ago
• iree-org / wave
  Wave: Python Domain-Specific Language for High Performance Machine Learning
  ☆42Updated this week
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆112Updated 7 months ago