https://github.com/hmunachi/nanodl

A Jax-based library for designing and training transformer models from scratch.
attention attention-mechanism deep-learning distributed-training flax gpt jax llama machine-learning mistral nlp transformer
Added: about 1 year ago - Last Synced: 11 months ago - Created: August 22, 2023

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits:
  • Committers:
  • Issues: 9
  • Pull Requests: 15
  • Owner: HMUNACHI
  • Stars: 256
  • Forks: 10
  • Packages: 1
  • Downloads: 103
https://github.com/ml4wifi-devs/mapc-sim

IEEE 802.11 MAPC (C-SR) simulator
coordinated-spatial-reuse jax simulator
Added: over 1 year ago - Last Synced: 11 months ago - Created: January 04, 2024

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 25
  • Committers: 4
  • Issues: 0
  • Pull Requests: 4
https://github.com/google/trax

Trax β€” Deep Learning with Clear Code and Speed
deep-learning deep-reinforcement-learning jax machine-learning numpy reinforcement-learning transformer
Added: over 1 year ago - Last Synced: 11 months ago - Created: October 05, 2019

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 1530
  • Committers: 81
  • Issues: 54
  • Pull Requests: 68
  • Owner: google
  • Stars: 7980
  • Forks: 814
  • Packages: 2
  • Downloads: 3,161
https://github.com/qazalbash/jaxtro

πŸ™ An open-source JAX-based gravitational-wave population inference πŸ”­
astrophysics gravitational-waves inference jax physics
Added: over 1 year ago - Last Synced: 11 months ago - Created: December 24, 2023

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 18
  • Committers: 1
  • Issues: 16
  • Pull Requests: 2
  • Owner: Qazalbash
  • Stars: 5
  • Forks: 0
  • Packages: 1
  • Downloads: 13
https://github.com/swanhubx/swanlab

🧐SwanLab: track and visualize all the pieces of your machine learning pipeline. 跟θΈͺδΈŽε―θ§†εŒ–δ½ ηš„ζœΊε™¨ε­¦δΉ ε…¨ζ΅η¨‹
data-science deep-learning fastapi jax machine-learning mlops model-versioning python pytorch tensorboard tensorflow tracking transformers visualization
Added: over 1 year ago - Last Synced: 11 months ago - Created: November 24, 2023

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 160
  • Committers: 11
  • Issues: 309
  • Pull Requests: 336
  • Owner: SwanHubX
  • Stars: 182
  • Forks: 25
  • Packages: 3
  • Downloads: 894
https://github.com/ami-iit/adam

adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
adam-robotics automatic-differentiation casadi jax motion-planning numpy optimization python pytorch reinforcement-learning rigid-body-dynamics robotics urdf
Added: over 1 year ago - Last Synced: 11 months ago - Created: May 25, 2021

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 332
  • Committers: 8
  • Issues: 30
  • Pull Requests: 54
  • Owner: ami-iit
  • Stars: 110
  • Forks: 17
  • Packages: 1
  • Downloads: 136
https://github.com/hilanzy/moss

Moss is a Python library for Reinforcement Learning.
atari dm-haiku dm-launchpad envpool impala jax library marl pettingzoo ppo pypi reinforcement-learning rlax
Added: over 1 year ago - Last Synced: 11 months ago - Created: February 19, 2023

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits:
  • Committers:
  • Issues: 47
  • Pull Requests: 92
  • Owner: hilanzy
  • Stars: 2
  • Forks: 0
  • Packages: 1
  • Downloads: 24
https://github.com/corl-team/xland-minigrid

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
jax meta-reinforcement-learning minigrid python reinforcement-learning xland
Added: over 1 year ago - Last Synced: 11 months ago - Created: December 01, 2023

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 33
  • Committers: 2
  • Issues: 4
  • Pull Requests: 1
  • Owner: corl-team
  • Stars: 142
  • Forks: 10
  • Packages: 0