69 Open source projects that are alternatives of or similar to jaxdf

Sparse Grid Discretization with the Discontinuous Galerkin Method for solving PDEs

Several code for the paper: A comparison study of deep Galerkin method and deep Ritz method for elliptic problems with different boundary conditions

ADAM implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Unofficial JAX implementations of deep learning research papers

PyFRAP: A Python based FRAP analysis tool box

Graphsignal Python agent

FEATool - "Physics Simulation Made Easy" (Fully Integrated FEA, FEniCS, OpenFOAM, SU2 Solver GUI & Multi-Physics Simulation Platform)

Source code for the examples accompanying the paper "Learning convex optimization control policies."

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Revisiting Rainbow

PIX is an image processing library in JAX, for JAX.

JAX implementations of core Deep RL algorithms

A low Mach number stellar hydrodynamics code

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

FuturEye_JIT is a Java Finite Element Method (FEM) library for solving PDE based forward/inverse problems as efficient as hand written FORTRAN code.

👩 Pytorch and Jax code for the Madam optimiser.

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

Mimetic Operators Library Enhanced

Trax — Deep Learning with Clear Code and Speed

Simple one-dimensional examples of various hydrodynamics techniques

Universal modeling and simulation of fluid dynamics upon machine learning

A small library for creating and manipulating custom JAX Pytree classes

Implementation of https://srush.github.io/annotated-s4

A GPT, made only of MLPs, in Jax

Factor graphs and nonlinear optimization for JAX

Documentation:

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

Computational Fluid Dynamics in JAX

Numerical Experiments

Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.

KoCLIP: Korean port of OpenAI CLIP, in Flax

A unified end-to-end learning and control framework that is able to learn a (neural) control objective function, dynamics equation, control policy, or/and optimal trajectory in a control system.

TCAD Semiconductor Device Simulator

Differentiating convex optimization programs w.r.t. program parameters

Chef Transformer 🍲 .

Multi-language suite for high-performance solvers of differential equations and scientific machine learning (SciML) components

ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)

A Julia package to perform Bifurcation Analysis

Benchmarks for scientific machine learning (SciML) software and differential equation solvers

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

A didactic Gaussian process package for researchers in Jax.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Visualize curve shortening flow in your browser.

Flax is a neural network library for JAX that is designed for flexibility.

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022

Python code for "Machine learning: a probabilistic perspective" (2nd edition)

Operator Inference for data-driven, non-intrusive model reduction of dynamical systems.

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

learning state-space targets in dynamical systems

Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)

Source code for the ICLR 2021 work "Generating Adversarial Computer Programs using Optimized Obfuscations"

SG⁺⁺ – the numerical library for Sparse Grids in all their variants.

A differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and discontinuities.

Functional models and algorithms for sparse signal processing

Federated posterior averaging implemented in JAX

A Python library for machine-learning and feedback loops on streaming data

JAX-based Spiking Neural Network framework