dm pixPIX is an image processing library in JAX, for JAX.
Stars: ✭ 271 (+118.55%)
lbfgsb-gpuAn open source library for the GPU-implementation of L-BFGS-B algorithm
Stars: ✭ 70 (-43.55%)
efficientnet-jaxEfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Stars: ✭ 114 (-8.06%)
ShinRLShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)
Stars: ✭ 30 (-75.81%)
GPJaxA didactic Gaussian process package for researchers in Jax.
Stars: ✭ 159 (+28.23%)
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 (+626.61%)
pdfoPowell's Derivative-Free Optimization solvers
Stars: ✭ 56 (-54.84%)
rA9JAX-based Spiking Neural Network framework
Stars: ✭ 60 (-51.61%)
mlp-gpt-jaxA GPT, made only of MLPs, in Jax
Stars: ✭ 53 (-57.26%)
galiniAn extensible MINLP solver
Stars: ✭ 29 (-76.61%)
jax-modelsUnofficial JAX implementations of deep learning research papers
Stars: ✭ 108 (-12.9%)
jax-cfdComputational Fluid Dynamics in JAX
Stars: ✭ 399 (+221.77%)
jax-rlJAX implementations of core Deep RL algorithms
Stars: ✭ 61 (-50.81%)
ADAMADAM implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
Stars: ✭ 51 (-58.87%)
snopt-matlabMatlab interface for sparse nonlinear optimizer SNOPT
Stars: ✭ 49 (-60.48%)
NAGPythonExamplesExamples and demos showing how to call functions from the NAG Library for Python
Stars: ✭ 46 (-62.9%)
MINLPLib.jlA JuMP-based library of Non-Linear and Mixed-Integer Non-Linear Programs
Stars: ✭ 30 (-75.81%)
omdJAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Stars: ✭ 43 (-65.32%)
graphsignalGraphsignal Python agent
Stars: ✭ 158 (+27.42%)
piperNo description or website provided.
Stars: ✭ 50 (-59.68%)
pytorchImproved LBFGS optimizer in PyTorch.
Stars: ✭ 16 (-87.1%)
dfognDFO-GN: Derivative-Free Optimization using Gauss-Newton
Stars: ✭ 20 (-83.87%)
NMPC-DCLF-DCBFA collection of work using nonlinear model predictive control (NMPC) with discrete-time control Lyapunov functions (CLFs) and control barrier functions (CBFs)
Stars: ✭ 38 (-69.35%)
car-racingA toolkit for testing control and planning algorithm for car racing.
Stars: ✭ 30 (-75.81%)
GALAHADA library of modern Fortran modules for nonlinear optimization
Stars: ✭ 60 (-51.61%)
articulated-pose[CVPR 2020, Oral] Category-Level Articulated Object Pose Estimation
Stars: ✭ 85 (-31.45%)
LBFGS-LiteLBFGS-Lite: A header-only L-BFGS unconstrained optimizer.
Stars: ✭ 98 (-20.97%)
Pajarito.jlA solver for mixed-integer convex optimization
Stars: ✭ 98 (-20.97%)
Transformers🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Stars: ✭ 55,742 (+44853.23%)
EinopsDeep learning operations reinvented (for pytorch, tensorflow, jax and others)
Stars: ✭ 4,022 (+3143.55%)
Thinc🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Stars: ✭ 2,422 (+1853.23%)
JaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Stars: ✭ 15,579 (+12463.71%)
FoolboxA Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Stars: ✭ 2,108 (+1600%)
FlaxFlax is a neural network library for JAX that is designed for flexibility.
Stars: ✭ 2,447 (+1873.39%)
TraxTrax — Deep Learning with Clear Code and Speed
Stars: ✭ 6,666 (+5275.81%)
PyprobmlPython code for "Machine learning: a probabilistic perspective" (2nd edition)
Stars: ✭ 4,197 (+3284.68%)
DatasetsTFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Stars: ✭ 3,094 (+2395.16%)
jax-resnetImplementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Stars: ✭ 61 (-50.81%)
treeoA small library for creating and manipulating custom JAX Pytree classes
Stars: ✭ 29 (-76.61%)
score flowOfficial code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
Stars: ✭ 49 (-60.48%)
annotated-s4Implementation of https://srush.github.io/annotated-s4
Stars: ✭ 133 (+7.26%)
SymJAXDocumentation:
Stars: ✭ 103 (-16.94%)
braxMassively parallel rigidbody physics simulation on accelerator hardware.
Stars: ✭ 1,208 (+874.19%)
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 (+84.68%)
bayexBayesian Optimization in JAX
Stars: ✭ 24 (-80.65%)
ML-Optimizers-JAXToy implementations of some popular ML optimizers using Python/JAX
Stars: ✭ 37 (-70.16%)
jaxdfA JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
Stars: ✭ 50 (-59.68%)
cr-sparseFunctional models and algorithms for sparse signal processing
Stars: ✭ 38 (-69.35%)
koclipKoCLIP: Korean port of OpenAI CLIP, in Flax
Stars: ✭ 80 (-35.48%)
madam👩 Pytorch and Jax code for the Madam optimiser.
Stars: ✭ 46 (-62.9%)
fedpaFederated posterior averaging implemented in JAX
Stars: ✭ 38 (-69.35%)
robustness-vitContains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).
Stars: ✭ 78 (-37.1%)
wax-mlA Python library for machine-learning and feedback loops on streaming data
Stars: ✭ 36 (-70.97%)