SymJAXDocumentation:
Stars: ✭ 103 (-74.19%)
braxMassively parallel rigidbody physics simulation on accelerator hardware.
Stars: ✭ 1,208 (+202.76%)
get-started-with-JAXThe purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
Stars: ✭ 229 (-42.61%)
bayexBayesian Optimization in JAX
Stars: ✭ 24 (-93.98%)
ML-Optimizers-JAXToy implementations of some popular ML optimizers using Python/JAX
Stars: ✭ 37 (-90.73%)
jaxdfA JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
Stars: ✭ 50 (-87.47%)
cr-sparseFunctional models and algorithms for sparse signal processing
Stars: ✭ 38 (-90.48%)
koclipKoCLIP: Korean port of OpenAI CLIP, in Flax
Stars: ✭ 80 (-79.95%)
madam👩 Pytorch and Jax code for the Madam optimiser.
Stars: ✭ 46 (-88.47%)
fedpaFederated posterior averaging implemented in JAX
Stars: ✭ 38 (-90.48%)
robustness-vitContains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).
Stars: ✭ 78 (-80.45%)
wax-mlA Python library for machine-learning and feedback loops on streaming data
Stars: ✭ 36 (-90.98%)
jaxfgFactor graphs and nonlinear optimization for JAX
Stars: ✭ 124 (-68.92%)
dm pixPIX is an image processing library in JAX, for JAX.
Stars: ✭ 271 (-32.08%)
efficientnet-jaxEfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Stars: ✭ 114 (-71.43%)
ShinRLShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)
Stars: ✭ 30 (-92.48%)
GPJaxA didactic Gaussian process package for researchers in Jax.
Stars: ✭ 159 (-60.15%)
uvadlc notebooksRepository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
Stars: ✭ 901 (+125.81%)
rA9JAX-based Spiking Neural Network framework
Stars: ✭ 60 (-84.96%)
mlp-gpt-jaxA GPT, made only of MLPs, in Jax
Stars: ✭ 53 (-86.72%)
jax-modelsUnofficial JAX implementations of deep learning research papers
Stars: ✭ 108 (-72.93%)