flax jax kolmogorov-arnold-networks

Engineering (40%)

flax flux jax pytorch variational-autoencoder variational-inference

blackjax colab flax jax jupyter-notebooks machine-learning numpyro pml probabilistic-programming pymc3 pyro pytorch tensorflow

deep-learning flax jax language-model machine-learning mnist optimization optimization-algorithms optimizer transformer

ai bayesian-inference calibration conformal-prediction deep-learning flax jax machine-learning ml model-calibration neural-networks uncertainty uncertainty-calibration uncertainty-estimation uncertainty-quantification

flax jax mox yax

deep-learning flax jax reinforcement-learning reinforcement-learning-algorithms reinforcement-learning-environments rl rl-environments

easylm flax gemma huggingface jax language-model tpu transformers

flax gnn graph-neural-networks jax

audio flax jax signal-processing

deep-learning direct-feedback-alignment feedback-alignment flax jax kolen-pollack