Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
Projects in Awesome Lists tagged with optax
A curated list of projects in awesome lists tagged with optax .
https://github.com/phlippe/uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
deep-learning flax jax optax pytorch pytorch-lightning tutorial tutorials uvadlc
Last synced: 09 Aug 2024
https://github.com/erfanzar/easydel
Accelerate, Optimize performance with streamlined training and serving options with JAX.
easydel flax gpt jax machine-learning mojo nlp optax transformers
Last synced: 07 Nov 2024
https://github.com/erfanzar/EasyDeL
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
easydel flax gpt jax machine-learning mojo nlp optax pytorch transformers
Last synced: 03 Aug 2024
https://github.com/borealisai/flora-opt
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
deep-learning flax jax large-language-models lora memory-efficient-tuning optax peft random-projection transformers
Last synced: 06 Nov 2024
https://github.com/frankroeder/goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
actor-critic deep-deterministic-policy-gradient deep-reinforcement-learning distrax droq flax goal-conditioned-rl goals gymnasium gymnasium-robotics jax mpi4jax optax reinforcement-learning robotics soft-actor-critic
Last synced: 05 Nov 2024
https://github.com/activatedgeek/optax-swag
Stochastic Weight Averaging (SWA) transforms for Optax with JAX
jax optax python stochastic-weight-averaging swag
Last synced: 14 Oct 2024
https://github.com/daniel-j-h/nedem
Neural implicit digital elevation model
dem digital-elevation-model flax glo-30 jax machine-learning optax
Last synced: 12 Oct 2024
https://github.com/bischtob/opterax
An gradient-free optimization suite written in JAX. We conform to the optax interface and provide ensemble-based optimizers.
jax machine-learning optax optimization
Last synced: 23 Oct 2024
https://github.com/jessefarebro/flax-mup
Maximal Update Parametrization (μP) with Flax & Optax.
Last synced: 14 Oct 2024
