#jax · dorkhub

Jax2D

Highly scalable 2D JAX physics engine.

Python★ 68

jax physics-engine physics-simulation

@MichaelTMatthews

vision

3D Vision functions with end-to-end support for deep learning developers, written in Ivy.

Python★ 72

3d deep-learning depth ivy jax machine-learning mxnet numpy

@ivy-llc

sax

S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.

Python★ 128

ann autograd circuit deep-learning jax optimization photonic-circuit photonic-optimization

@gdsfactory

jax-am

Additive manufacturing simulation with JAX.

Jupyter Notebook★ 360

additive-manufacturing cfd crystal-plasticity finite-element-method jax phase-field-method simulation

@CMSL-HKUST

netket

Machine learning algorithms for many-body quantum systems

Python★ 693

deep-learning hamiltonian jax machine-learning markov-chain-monte-carlo monte-carlo-methods neural-networks neural-quantum-states

@netket

PDEBench

PDEBench: An Extensive Benchmark for Scientific Machine Learning

Python★ 1.2k

ai autoregressive-models benchmark deep-learning fluid-dynamics jax machine-learning navier-stokes-equations

@pdebench

skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/G…

Python★ 1.1k

brax deep-learning flax gym gymnasium isaaclab isaacsim jax

@Toni-SM

brax

Massively parallel rigidbody physics simulation on accelerator hardware.

Jupyter Notebook★ 3.2k

jax physics-simulation reinforcement-learning robotics

@google

jax-js

JAX in JavaScript – ML library for the web, running on WebGPU & Wasm

TypeScript★ 888

javascript jax jit machine-learning neural-networks numpy wasm webgl

@ekzhang

DawDreamer

Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX, Warp Markers, and JUC…

C++★ 1.3k

ableton audio audio-plugin audio-processing daw faust jax juce

@DBraun

search