Version 0.0.40

What's Changed

• Updating JAXBeta branch from main branch by @erfanzar in #42
• Update Beta Branch by @erfanzar in #48
• Update V0.0.40 Beta (Adding Flash Attention, Adding 8,6,4 Bit models ,improving Documentations) by @erfanzar in #52
• Fix eval batch loop (beta branch) by @w11wo in #51
• Support Sphinx Docstring Format by @w11wo in #53
• Update Beta Branch by @erfanzar in #55
• Update Beta Branch by @erfanzar in #56
• Updating Mistral and Llama Models by @erfanzar in #57
• Updating Beta Branch by @erfanzar in #58
• Changing Mesh by @erfanzar in #60
• Updating Beta by @erfanzar in #62
• 4D Mesh now supported for all the Models and BITs improved by @erfanzar in #64

Full Changelog: 0.0.38...0.0.40