EnzymeAD / Enzyme-JAX

Related projects ⓘ

Alternatives and complementary repositories for Enzyme-JAX

• PhilipVinc / Jax.jl
  ☆29Updated 2 years ago
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆80Updated this week
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated 2 months ago
• ranjanan / MonteCarloIntegration.jl
  A package for multi-dimensional integration using monte carlo methods
  ☆38Updated 9 months ago
• JuliaIPU / IPUToolkit.jl
  IPU programming in Julia
  ☆30Updated last month
• YichengDWu / MoYe.jl
  Programming Gemm Kernels on NVIDIA GPUs with Tensor Cores in Julia
  ☆36Updated this week
• Argonne-National-Laboratory / Checkpointing.jl
  Checkpointing for Automatic Differentiation
  ☆52Updated 2 weeks ago
• EnzymeAD / Reactant.jl
  ☆67Updated this week
• JuliaDiff / TaylorDiff.jl
  Taylor-mode automatic differentiation for higher-order derivatives
  ☆75Updated this week
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• JuliaLabs / MLIR.jl
  ☆56Updated last week
• nathanaelbosch / Fenrir.jl
  Physics-Enhanced Regression for Initial Value Problems
  ☆19Updated 9 months ago
• SciML / DeepEquilibriumNetworks.jl
  Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.
  ☆50Updated 2 weeks ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆10Updated 3 weeks ago
• JuliaLabs / LULESH.jl
  Julia implementation of LULESH with MPI + X.
  ☆13Updated 2 years ago
• omlins / libdiffusion
  Proof of Concept: a C-callable GPU-enabled parallel 2-D heat diffusion solver written in Julia using CUDA, MPI and graphics
  ☆24Updated 4 years ago
• FluxML / XLA.jl
  "Maybe we have our own magic."
  ☆47Updated 4 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆81Updated 2 weeks ago
• YichengDWu / NeuralGraphPDE.jl
  Integrating Neural Ordinary Differential Equations, the Method of Lines, and Graph Neural Networks
  ☆18Updated last year
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated last year
• odsl-team / julia-ml-from-scratch
  Machine learning from scratch in Julia
  ☆31Updated last month
• avik-pal / FluxMPI.jl
  Distributed Data Parallel Training of Deep Neural Networks
  ☆56Updated 8 months ago
• JuliaLabs / brutus
  ☆60Updated 10 months ago
• Pangoraw / Coil.jl
  ꩜ Lift Julia array operations to MLIR dialects and run using IREE.
  ☆42Updated 10 months ago
• utkarsh530 / GPUODEBenchmarks
  Comparsion of Julia's GPU Kernel based ODE solvers with other open-source GPU ODE solvers
  ☆24Updated 10 months ago
• rejuvyesh / PyCallChainRules.jl
  Differentiate python calls from Julia
  ☆56Updated 2 years ago
• rakeshvar / Zygote-Mutating-Arrays-WorkAround.jl
  A tutorial on how to work around ‘Mutating arrays is not supported’ error while performing automatic differentiation (AD) using the Julia…
  ☆30Updated 3 years ago
• SciML / OperatorLearning.jl
  No need to train, he's a smooth operator
  ☆43Updated last week
• TuringLang / NestedSamplers.jl
  Implementations of single and multi-ellipsoid nested sampling
  ☆43Updated last month
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆34Updated 3 weeks ago