A JAX implementation of DEformer for arbitrary conditioning.
-
Updated
May 29, 2022 - Python
A JAX implementation of DEformer for arbitrary conditioning.
A generalized implementation of Grad-CAM for Flax
minimal forward-mode automatic differentiation using python's abstract syntax tree
Common practices for distributed training using various backends
Conversion of LAST library from JAX to PyTorch
NXML is an eXtension for Machine Learning
implicit differentiation with jax
Denoising diffusion operators in JAX/FLAX
API for recipe generation from flax-community/t5-recipe-generation (huggingface)
msThesis
A fork of a library in JAX for model parallelism of transformers, for "Project Cybertard"
A Multivariate Gaussian Bayes classifier written using JAX
Image Quality Assessment library for jax. Converts the Numpy implementations from BasicSR into Jax.Numpy.
Pure-Python, JAX-accelerate implementation of the pyloric circuit model described by Prinz et al (Nat. Neurosci., 2004)
An implementation of WideResNets with Fixup initialization in Jax/Flax. This can be useful for use cases where Batch Normalization should be avoided (for example when using the Laplace approximation to the Bayesian posterior).
JAX and numba implementations of the fast upper envelope scan for solving discrete-continuous dynamic programming problems
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."